在windows下用anaconda虚拟环境安装pytorch gpu版 成功!
文章目录
接下来介绍安装过程。
推荐全部看完本文再进行安装
可以重点看黑体字,0部分介绍pip安装,后面的部分介绍conda安装,pip部分简明扼要,conda部分介绍的更详细些。祝你安装成功。
0. 用pip安装
pip比conda 速度更快一些,更随心所欲一些,所以介绍一下pip安装,直接上代码
conda create -n pytorch18 python=3.8 # 创建pytorch的虚拟环境
activate pytorch18 # 激活pytorch环境
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
# 上面的安装代码去https://pytorch.org/get-started/previous-versions/找,根据自己想用的pytorch版本和cuda版本选择相应的命令。关于如何查看cuda版本可以在我目录中的第三部分去查看。
# 安装过程会发现安装很慢,这时候ctl+c取消安装,找到这个下载速度很慢的安装包的来源网址,这个网址在安装界面中会显示,复制,去浏览器中打开,然后crl+f,输入安装包的名字,可以找到需要的安装包,然后直接离线下载到本地。
pip install E:\pytorch\torch-1.8.0+cu111-cp38-cp38-win_amd64.whl # 后面的路径就是安装包在的路径,直接安装
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
# 再次输入这个命令来安装另外两个模块。搞定
pip install ipython # 安装ipython测试是否安装成功
ipython # 打开ipython
import torch
torch.ones(8) # 创建一个tensor
torch.cuda.is_available() # 测试gpu是否可以使用,成功的话会返回True
1. 安装准备
1.是windows系统下的安装
2.其次是在anaconda的虚拟环境中安装,anaconda3的安装以及如何创建虚拟环境就不讲了。
使用conda前在base环境下进行大扫除:
清除下缓存信息
conda clean --packages --tarballs
2. 安装指令
3.pytorch官网(官网链接)讲了如何用conda安装。如下图,,如果你想安装其他版本,也可以点击图片中右上角的install previous versions of pytorch
输入蓝框中的pytorch安装指令进行安装。
需要选择cuda版本,那么如何查看cuda的版本呢?
3. 查看、提升cuda版本
4.nvidia显卡是否支持cuda?如何查看cuda版本?如何提升cuda版本?
3.1 判断显卡是否支持cuda
下图显示了支持cuda的显卡
3.2 查看 cuda版本
桌面右击打开nvidia控制面板,点击左下角系统信息,点击组件,在3D设置里的第三个就是cuda版本了,我的是8.0.0。
3.3 提高cuda版本
驱动版本和cuda版本是有关系的,因此下载更高版本的驱动程序就可以得到更高版本的cuda。下图显示了驱动和cuda版本之间的关系。
3.4 查看驱动版本
如下图,在nvidia控制面板的系统信息中,可以查看驱动程序版本。
nvidia官网没有旧的驱动程序下载,可以到这个网站https://down.gamersky.com/oth/soft/下载。
4. 离线下载torch安装包
现在可以输入pytorch安装指令进行安装了。但是下载很慢。为了避免连接中断导致下载失败,所以离线下载torch安装包,分两步操作:
4.1 离线下载安装包
输入pytorch安装指令后,会显示各个包的下载地址(如下图中蓝框)地址显示的pytroch指的就是https://conda.anaconda.org/pytorch/win-64/这个网站。直接打开这个网址下载pytorch安装包。迅雷两分钟搞定。
与此同时我们需要输入ctrl+c来关闭刚刚进行的安装,我们只是为了看下pytorch安装包的下载网站。接着要输入下列指令,来清除刚刚下载所产生的缓存文件。
conda clean --packages --tarballs
安装包下载下来之后,将pytorch安装包复制到 \Anaconda3\pkgs 中去。
4.2 修改urls.txt文件
第二步:上图中最下面有一个urls.txt文件,将迅雷中pytorch安装包的下载链接复制到urls.txt文件中。
从下图可以看到我复制在了第一行。这样pytorch安装包就不需要下载了
5. 安装及检查
输入pytorch安装指令,回车。
检查是否安装成功,同样pytorch官网也给出了方法如下图所示:
恭喜你,pytorch安装成功!有问题私信或留言。