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

pytorch gpu版本安装(windows & anaconda3)

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

pytorch gpu版本安装(windows & anaconda3)


pytorch安装成功!

接下来介绍下安装过程。
推荐全部看完本文再进行安装

1.介绍的是windows系统下的安装

2.其次是在anaconda的虚拟环境中安装,anaconda3的安装以及如何创建虚拟环境就不讲了。
使用conda前先在base环境下进行下大扫除:
首先清除下缓存信息,更新下包

conda clean --packages --tarballs
conda update --all

看看风景放松下,把脚伸水里,享受着海风,慢慢看。

3.pytorch官网(官网链接)讲了如何用conda安装。如下图

选择符合情况的选项后,在新建的pytorch虚拟环境中输入蓝框中的pytorch安装指令进行安装。
上面需要选择自己的cuda版本,那么如何查看自己cuda的版本呢?

4.怎么知道nvidia显卡是否支持cuda?如何查看cuda版本?如何得到我想要的cuda版本?
第一个问题,判断显卡是否支持cuda:
下图显示了支持cuda的显卡

第二个问题,查看 cuda版本:
桌面右击打开nvidia控制面板,点击左下角系统信息,点击组件,在3D设置里的第三个就是cuda版本了,我的是8.0.0。

如何提高cuda版本:
驱动版本和cuda版本是有关系的,因此下载更高版本的驱动程序就可以得到更高版本的cuda。下图显示了驱动和cuda版本之间的关系。

如何查看驱动版本呢?在nvidia控制面板的系统信息中,可以查看驱动程序版本。

现在你可以输入pytorch安装指令进行安装了。但是会遇到了一个新问题,pytorch安装包有500兆左右,输入安装指令后,下载很慢。

6.使用清华镜像源之后我的下载速度还是很慢。所以我删除了所有镜像源(记得删掉镜像源之后用 conda clean -i 指令刷新下),选择直接到官网下载安装包。具体操作如下
第一步:
输入pytorch安装指令后,会显示各个包的下载地址,地址显示的pytroch指的就是https://conda.anaconda.org/pytorch/win-64/这个网站。我们要做的就是直接打开这个网址下载pytorch安装包。我使用迅雷下载,一分钟就下完了。
下面这张图就显示了pytorch的下载网站是pytorch,也就是https://conda.anaconda.org/pytorch/win-64/

与此同时我们可能需要通过ctrl+c来关闭刚刚进行的安装,因为我们只是为了看下pytorch安装包的下载网站。因此接着要输入下列指令,来清除刚刚下载所产生的缓存文件。这里我不确定不同的环境下使用指令会不会效果不一样,推荐去base环境下使用。

conda clean --packages --tarballs

安装包下载下来之后,将pytorch安装包复制到 Anaconda3pkgs 中去。

第二步:这步最重要,上图中最下面有一个urls.txt文件,将迅雷中pytorch安装包的下载链接复制到urls.txt文件中。

可以看到我复制在了第一行。这样我们的pytorch安装包就不需要再下载了,也不会存在下载速度慢的问题了。

7.输入pytorch安装指令,回车,然后愉快的安装吧,不必再忍受等待的煎熬了。

7.检查是否安装成功,同样pytorch官网也给出了方法

恭喜你,pytorch安装成功!

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

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

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