在windowws下用anaconda安装 pytorch

概述

本次主要是谈论下如何在windows10下面安装pytorch的gpu版本,第一次写博客,可能写的有点啰嗦,还望大家耐心看下去。

1.安装anaconda

这个网上的教学很多,百度一下即可,要注意的是由于外网不科学上网速度贼慢,需要添加清华园,在命令行(win+r再输入cmd)或者anaconda prompt中输入如下配置。如果已经装过anaconda但是在安装时出现环境之类的错误时,建议卸掉重新安装,我当时出现了这种问题重新安装便解决了。

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

2.查看自己的显卡驱动版本(很重要!!!)

之前大概是因为版本不匹配,我安装的gpu版本的pytorch,但是gpu无效!打开控制面板,点击卸载程序,查看NVIDIA的版本,下图中可以看出我的版本是441.66。

在这里插入图片描述
然后打开 这里 在网页中下拉,查看表格,找到适合自己型号的CUDA版本,由于我是重新安装的最新版本(2019年1月26日),且是windows10环境,版本为441.66,满足最右列的所有值,所有版本的CUDA我均可以安装,但是注意,anaconda会自动帮我们安装CUDA Toolkit,我们不需要安装CUDA和CUDNN!这里需要记住能用哪些版本的CUDA,后面要用到!
在这里插入图片描述

可能遇到的问题

你可能会发现你的显卡驱动,版本很低,并且低版本可能会在后续的安装中出现很多很多的问题,比如版本不匹配,国内源不支持等等问题。如果出现了,建议卸载显卡驱动,安装较新版本的驱动。这里百度即可,需要注意的是卸载后需要重启电脑,否则会无法安装。下面主要谈谈一些重要的方面。

显卡型号

右击此电脑,点击管理,点击设备管理器,点击显示配置器,下图可以看到,我的显卡型号是NVIDIA GeForce GTX 1050

在这里插入图片描述
点击 这里 ,安装刚刚找到的显卡型号,搜索你所需要的驱动型号,点击search。这里Download Type的Studio Driver是时候我们学习办公使用的,而Game Ready Driver则是适合游戏的,这里我选择的Studio Driver。
在这里插入图片描述

这是我搜索到的结果,点击下载即可,然后就按照网上我教学安装吧。

在这里插入图片描述

3.安装pytorch

登录官网 这里 ,下拉网页,按照电脑的配置选择版本。我这里是选择的stable稳定版本,windows,用conda安装,py是3.7版本,CUDA前面查到了是10.2以下都可以用,最后得到了cmd下的运行语句 conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

需要注意的是最后运行的时候要去掉最后的-c pytorch,这代表从官网下载,而不是国内源。
所以运行下面语句

conda install pytorch torchvision cudatoolkit=10.1

在这里插入图片描述

注意点

一定要去掉最后的-c pytorch,当然如果你能fq另当别论。此外,需要注意版本的选取,建议选取运行语句最后只有-c pytorch的,而没有别的内容的,比如下图的内容多出了-c defaults和-c numba/label/dev,这里当时对我造成了很大的困扰,因为国内源没有某些资源,下载速度慢的让人抓狂,并且最后成功安装后gpu无效,可能是我当初没有重装显卡驱动的缘故。总之这样之后最后的安装时间不到二十分钟。
在这里插入图片描述

判断是否安装成功

运行如下代码,返回True则表示gpu版本运行成功,False则是gpu无法使用,但是torch照常能用,就会后面操作可能会很慢。

import torch
print(torch.cuda.is_available())

在这里插入图片描述
运行成功!

总结

第一次写博客,还望多多包涵。

发布了1 篇原创文章 · 获赞 1 · 访问量 38

猜你喜欢

转载自blog.csdn.net/qq_40589781/article/details/104091782