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(无需换源)

二、NVIDIA驱动安装与更新

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

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

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

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

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

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

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

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

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

四、下载pytorch

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

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

第一个选择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(无需换源)

Original: https://blog.csdn.net/m0_55216608/article/details/123138069
Author: 丿Qls丶
Title: anaconda安装pytorch-gpu(无需换源)

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/712769/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球