anaconda安装pytorch-gpu(无需换源)

如何在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

得到如图:

anaconda安装pytorch-gpu(无需换源)_第1张图片

二、NVIDIA驱动安装与更新

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

anaconda安装pytorch-gpu(无需换源)_第2张图片

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

anaconda安装pytorch-gpu(无需换源)_第3张图片

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

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

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

anaconda安装pytorch-gpu(无需换源)_第4张图片

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

 四、下载pytorch

进入pytorch的官网,下翻后得到如图:

anaconda安装pytorch-gpu(无需换源)_第5张图片

       第一个选择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的版本。 

anaconda安装pytorch-gpu(无需换源)_第6张图片

你可能感兴趣的:(笔记,anaconda,pytorch,深度学习)