在pytorch中使用GPU

今天学习的时候突然发现我的torch.cuda.is_available()返回的是False。这不能忍，我可是高贵的3060用户啊。得改。首先查看我使用的是11.4版本的CUDA

然后查看我的pytorch版本

原来我装的是cpu版本的torch。那去官网看看我应该下载哪个版本的。Previous PyTorch Versions | PyTorch

看了看之前版本最高支持到11.1的，那看看最新版本的。torch-1.10，能支持CUDA 11.3的，就决定是你了。去吧conda，使用

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

 

 遇到了一些问题。

 复制粘贴到搜索引擎，在 installation - PackagesNotFoundError: The following packages are not available from current channels: - Stack Overflow提示我使用把conda-forge加入channels。然后就可以了。

conda config --append channels conda-forge

了解一下原因，答主说channels是存放packages的服务器，而从社区维护的conda-forge启动在标准的channels不可用时是一个很好的方法。

如果一些包在windows上不被支持，那可能就只能用pip安装了。这样还是会安装在你的conda目录下，但是是使用python内置的包管理工具。有时候有些包只能用pip下载。

其中最后的-c代表的是channels，从pytorch的官网下载，这会导致速度很慢。所以我们要从国内的镜像源找我们需要的torch包。例如清华源。添加到conda源中。

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

但据说清华源也不稳定，所以看个人了。不然就去pytorch官网下载下来离线安装。

最后结果检查。

 下班吃宵夜去了。