在官网下载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的时候报的错
解决方案 重装pyzmqpip uninstall pyzmq # 卸载pyzmq pip install pyzmq==19.0.2 #安装pyzmq19.0.2版本Pytorch
装pytorch不需要另外去匹配CUDA的版本与Python的版本,只有机子里面有CUDA一般都会有,就不用另外装,cudatoolkit这玩意儿估计能解决版本的问题,我用的就是python3.6然后环境里面是9.0的CUDA,也能照常使用



