核心的步骤按这篇文章:
Python深度学习环境配置(Pytorch、CUDA、cuDNN),包括Anaconda搭配Pycharm的环境搭建以及基础使用教程(保姆级教程,适合小白、深度学习零基础入门)_python torch cuda-CSDN博客
一些可能出错的地方解决方法:
cuda的安装最好是默认地址不要改
下载pytorch时最好不要用清华源,使用本地下载最好(附地址如下)
清华源应该是只能映射CPU版本,想要下载GPU版本不能使用清华源,
所以可以下载到本地然后再安装
一个是pytorch一个是torchversion,两个都下
download.pytorch.org/whl/torchvision/
download.pytorch.org/whl/torch/
具体实现本地安装方法见此文:
安装pytorch(下载到本地进行简单安装)_pytorch本地下载-CSDN博客
查看自己电脑应该装哪个版本cuda和pytorch看这篇:
安装pytorch(下载到本地进行简单安装)_pytorch本地下载-CSDN博客