提示:惠普暗影精灵2+win10+GTX1050用conda安装PytorchGPU环境。
conda下载安装完成之后,你在开始菜单里面找会找到anconda3的文件夹,里面会有anaconda promt(这是个黑窗不要害怕,我一开始以为是什么病毒,着实吓一跳。)
打开promt之后一开始是base路径
conda info -e
会显示就一个环境并且,*标记就在base里面。
conda create -n pytorchgpu python=3.7
-n 后面跟的就是你的环境的名字,记住它之后还要用。python=3.7是环境python语言的版本。
你可以在用之前查看环境信息的方式查看,现在就会有两个环境名,*还是在base环境前面。
conda activate pytorchgpu
正常来说你的黑窗显示的应该是:
(pytorchgpu)c:/xxxx/xxxx/
这样就打开了这个环境就可以开始安装库了。
pytorch官网 pytorch.org 可以查看你需要的安装版本的conda指令。这个需要自己查阅。如果你跟我是一个配置,可以用下面这条:
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch
注意了这里的-c pytorch意思是从pytorch官网的数据库下载,我之前不是说英伟达在国内下载很慢么?这个pytorch的服务器和anaconda的服务器也是一样,都很慢。
但是有些地区不一样,我也不太明白其中的道理。有些地方他就是下的快。
正常来说,他要是能下是最好的,你等他下载就完事了。很慢就接着看下面。
conda在安装包时,是由软件内部通过url请求数据库地址来下载的。这个数据库就是常说的源。国内一些高校和公司也是要用外国的数据库的,但是,我们国家有墙(你懂的)。那就只能镜像服务器(就是拷贝人家的数据库回来换国内的服务器。)这里只介绍清华源,其他的源自己开发吧,常见的还有中科大 豆瓣 上海交大。。。
还是在刚刚那个黑窗。
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
一般这三个源就够了。不同的源修改网址就够了。
步骤操作还是之前那个黑窗->激活环境->安装但是要记住去掉-c pytorch
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0
详情看这个大佬写的帖子。传送门
写的很好。
大的安装包 比如cudatoolkit pytorch 确实很大。可以用一般的网把小的安装了。 在安装时他会写出来什么包是在哪个网址安装的,把那个网址记住 后面跟上你要安装的包名就是这么一个链接。打开迅雷,新建一个项目用url直接下载。
比如说我下载的pytorch
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/pytorch-1.1.0-py3.7_cuda90_cudnn7_1.tar.bz2
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64这个就是数据库网站。
pytorch-1.1.0-py3.7_cuda90_cudnn7_1.tar.bz2这个就是包名。
合起来复制用迅雷新建项目,就可以直接下载。
下载完毕之后,还是打开promt 激活环境然后cd命令进入你迅雷下载的目录
比如
(pytorchgpu)c:/xxxx/xxxx/cd\
(pytorchgpu)c:/e:/xxx/xxx
(pytorchgpu)e:/xxx/xxx/
我写的cd命令可能不对,这个东西也是上网查一下就知道了。
然后就用 如下命令安装。
pip install <包名>
比如:
pip install --offline pytorch-1.1.0-py3.7_cuda90_cudnn7_1.tar.bz
pip会比较快,一般conda创建环境都会有pip工具。
还有一点在你第一次安装一些小一点的包的时候,有大的包,比如 pytorch 没有安装完。第二次安装 他会提示你warning告诉你说有东西需要手动请理。把那个截图或者复制记下来。按照他的提示把包请理干净再用离线的方法装。
安装完了之后可以直接用黑窗打开环境 然后python 进入解释器输入如下代码。
import torch
torch.cuda.is_available()
#cuda是否可用;
torch.cuda.device_count()
#返回gpu数量;
torch.cuda.get_device_name(0)
#返回gpu名字,设备索引默认从0开始;
torch.cuda.current_device()
#返回当前设备索引;
希望能给还没有安装好pytorch或者被pytorch安装很慢所苦恼的xdm给点帮助。不要像我装了3天多。总算摸清楚。
补:换源
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/pkgs/main
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/pkgs/free
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/pkgs/r
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/pkgs/pro
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/pkgs/msys2
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/cloud/conda-forge
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch