版本 anaconda3.0 cuda10.2 cudnn python=3.9 torch1.8.0+torchvision0.9.0
-
anaconda3 +cuda10.2+ cudnn+pycharm 官网下载,比较容易
-
选择对应cuda10.2版本的torch1.8.0+torchvision0.9.0 pytorch轮子文件
-
清华源镜像gpu版本会出现下载的是cpu版本(花了我两天时间,看了一个老哥的博客我才知道)
-
torch1.9.0+torchvision0.10.0可能会出现(果然是这版本出现的问题)
RuntimeError: Couldn't load custom C++ ops. This can happen if your PyTorch and torchvision versions are incompatible, or