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

得到如图:

二、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的版本。 

  • 7
    点赞
  • 19
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值