栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

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

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

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

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/753023.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号