项目场景:
安装Pytorch,安装中出现的问题
在官网下载pytorch
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
第一个BUG:Key already registered with the same priority: GroupSpatialSoftmax
安装完进入python,输入以下代码
import torch
会报如下错误Key already registered with the same priority: GroupSpatialSoftmax
解决方案:
卸载两次pytorch,再卸载一次,再重装(不会真正的重装,会有缓存)
pip uninstall torch
pip uninstall torch
conda uninstall pytorch
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
再次进入python运行以下代码
import torch
torch.cuda.is_available()
>>> True
第二个BUG:Failed to start the Kernel
这是用vscode打开装了ipykernel的jupyter notebook的时候报的错
解决方案 重装pyzmq
pip uninstall pyzmq
pip install pyzmq==19.0.2
Pytorch
装pytorch不需要另外去匹配CUDA的版本与Python的版本,只有机子里面有CUDA一般都会有,就不用另外装,cudatoolkit这玩意儿估计能解决版本的问题,我用的就是python3.6然后环境里面是9.0的CUDA,也能照常使用
|