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

只需两步,在anaconda里安装GPU版的pytorch

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

只需两步,在anaconda里安装GPU版的pytorch

1.主要是显卡驱动的版本,尽量更新到最新版(更新到最新就不要考虑驱动版本和cuda版本的对应,会方便很多)

驱动版本查看:输入命令nvidia-smi,如图

如果出现以下提示:

’nvidia-smi’ 不是内部或外部命令,也不是可运行的程序

参考这里https://blog.csdn.net/shuiyixin/article/details/99897640

显卡驱动更新教程:https://zhidao.baidu.com/question/1951584892469977108.html

显卡驱动与cuda的版本对应图:

2.需要安装的包主要包括三个:pytorch,cudatoolkit,cudann(这三个包同样要注意版本对应) torchvision和torchaudio按需安装。

清华镜像设置:进入C:UsersAdministrator,找到.condarc文件,粘贴我下面的代码覆盖原来的

ssl_verify: true
show_channel_urls: true
channels:
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/

下载三个主要包的命令:

conda install pytorch=你所需版本号
conda install cudatoolkit=你所需版本号
conda install cudann=你所需版本号

cudatoolkit与pytorch的版本对应:可以看pytorch的版本号,例如pytorch=1.7.1+cu101代表对应cudatoolk10.1版本的pytorch1.7.1。有点拗口,懂就行。

cudatoolkit与cudann的版本对应:在安装完pytorch和cudatoolkit后可以输入命令conda search cudnn --info有版本的对应信息,如下图:

最后附上pytorch官网:https://pytorch.org/

搞了一天,最后原来是更新个显卡驱动这么简单的事。。。

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

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

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