1.使用清华源搭建pytorch环境过程。
首先说明本文是在windows环境下实现的。因为在使用pytorch官网的连接下载时网速太慢导致时间过长断开连接所以使用了清华的源来下载,亲测速度相当的快。打开安装的anaconda prompt中默认进入了base环境,首先是下面的几行代码,注意不要搞错顺序
下面展示一些 内联代码片
。
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
然后结束环境的下载了,自己并没有在base环境下安装pytorch,而是创建了一个基于python 3.8版本的pytorch环境,方便以后切换不同的环境。使用下面的代码创建环境。
conda create -n pytorch python=3.8
这时候就发现下载的速度非常的快。
然后使用下面的代码进入自己创建的pytorch环境
conda activate pytorch
最后是从pytorch官网选择自己要用的配置,复制官网让输入的那行代码,要把最后的-c pytorch去掉就行。
conda install pytorch torchvision torchaudio cudatoolkit=10.2
因为自己下载时官网给的最低版本是10.2,所以就选择了这个。
下面强调一点
就是如何选择自己需要的cuda,其实我觉得要是选择对了(兼容),也不会出现torch.cuda.is_available()为false的情况。
2.torch.cuda.is_available()为false的解决方案
我的是nvidia的独显,要是报错了的话,先去查看一下自己的gpu型号支不支持cuda,支持的话,再更新一下自己gpu的驱动吧。去nvidia官网就可以下载,在控制面板里面可以查看自己的gpu型号。
ps:我的是这样成功的,别的方法咱也不会了,哈哈哈。