如何在anaconda里面安装gpu版本的pytorch?

注：本人翻看了大量的帖子并参照了很多方法去安装，最后发现输入print(torch.cuda.is_available())的时候返回的总是false，原因可能是很多人因为pytorch官网下载很慢，然后去选择换源，然后conda方式的安装，其他源中很可能会出现没有对应pytorch的cuda，而且anconda最近默认下载

一、打开anaconda prompt输入

conda create -n pytorch python=3.7

再输入activate pytorch来激活环境

activate pytorch

得到如图：

二、NVIDIA驱动安装与更新

       1.去查看自己电脑的显卡版本，右击此电脑-->管理-->设备管理器-->显示适配器。就可以看到电脑显卡的版本了。如图，可以看到我的是一块RTX1650ti显卡。

2.去英伟达官网下载对应版本的驱动官方驱动 | NVIDIA

​

       选择对应的产品类型、系列、家族、操作系统，下载类型里面有两个，两个都可以选择，语言选择Chinese（simplified） 。

三、查看电脑支持的cuda版本

打开anaconda prompt输入nvidia-smi得到：

​

    如图可知自己的驱动版本和cuda的最高支持版本（cuda可以向下兼容），我的是驱动版本是511.65，cuda版本是11.6，说明最高支持11.6版本的cuda。

 四、下载pytorch

进入pytorch的官网，下翻后得到如图：

       第一个选择stable,第二个根据自己的需要来选，第三个一定要选pip，然后根据电脑支持的cuda去选择cuda的版本，这个里面有其他版本的cuda（要注意自己虚拟环境的python版本），最后复制run this command 里面的指令到anaconda prompt去执行。

 五、检验是否成功安装pytorch

输入python出现>>>后输入import torch回车，输入 print(torch.cuda.is_available())如果返回true说明下载成功，然后可以输入print(torch.cuda_version)来查看cuda的版本。