博客

Windows下安装pytorch-cpu版本后，再安装gpu版本

由于已经安装了cpu版本了，如果再在该环境下安装gpu版本会造成环境污染，因此，再安装gpu版本时，需要再新建一个虚拟环境才能安装成功。

1.首先安装cuda


在安装之前应该先查看一下自己所适配的版本



打开系统信息，点击组件可以查看到

然后去官网下载所适配的版本。

这里可以下载到自己电脑所适配的NVIDIA：

此处



接下来的步骤网上很多博客都写了，如果自己的网络不佳，记得选择local版本的下载，否则很容易出错哈。


此处博主写的就很好


安装完cuda和cudnn后，开始安装pytorch的gpu版本。


2.安装gpu版本


打开anaconda prompt，首先创建虚拟环境：

#创建新环境
conda create --name pytorch-gpu python=3.7
#激活环境
conda activate pytorch-gpu

输入上面两个语句，名字可以随意取。

激活后去pytorch官网

链接

选择适合自己的版本，但是使用conda方式下载，直接用这个语句是不能下载成功的，因为它的服务器在国外，所以要选择下国内的镜像源。


参考该博主

# 添加Anaconda的清华镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
# 设置搜索时显示通道地址
conda config --set show_channel_urls yes  

输入上述语句，后输入


conda install pytorch torchvision cudatoolkit=10.2


此处记得把-c pytorch去掉。就可以直接下载了。安装过程中可能cudatoolkit=10.2这个会出现中断，只需要多数入几次conda install pytorch torchvision cudatoolkit=10.2即可！

期间遇到了warning：A newer version of conda exists。如下图所示：



方法：

conda install conda=4.8.3

这里根据自己最新版本进行修改，可能一次弄会失败，我就失败了一次，第二次就成功了。

然后查看

conda info

进入pycharm选择interpreter


1.选择file-settings

2.选择project-interpreter

3.选择下图中右边的设置，点ADD



剩下的可以参考

该博主


最后总结下，太心累了