1、pytorch下载
打开网站https://pytorch.org/,会自动根据电脑配置给出下载命令,我的是:
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
2、添加镜像
由于网络问题,直接运行上面命令无法下载,出现http错误的问题,需要添加镜像。
添加镜像后,在c:\Users\用户名 这个目录下生成.condarc文件,即配置的镜像路径。
根据博客https://blog.youkuaiyun.com/jiyangsb/article/details/82430794,配置镜像后,要将上面的conda安装命令后面的-c pytorch去掉,不然还是按默认路径安装。
(1)据说中科大的比较快
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
但是这个镜像没有我要下载的版本
(2)清华的镜像
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/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
但是,我添加了之后,仍然http错误。
解决方法:把.condar文件default那一行删除。
解决之后,发现清华镜像仍然没有我要的版本。。。
然后我删掉了.condar文件(也就是不用镜像),继续执行安装命令:
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
可能是因为比较晚,网速变快了,这次居然没有出现http错误,就是超级慢。。。
该命令安装了cudatoolkit=9.0、pytorch、torchvision三个内容
总结:如果cuda版本比较旧,使用镜像安装更快。如果镜像中没有需要的版本,找一个网速好的时间段再下载吧。。
3、测试安装
import torch
if __name__ == "__main__":
print(torch.__version__)
print(torch.cuda.is_available())
输出:
1.0.1
True
Pytorch位置:
参考:
1、pytorch在windows10上安装使用:https://blog.youkuaiyun.com/cuixing001/article/details/81952116
2、贾扬清知乎回答: https://www.zhihu.com/question/59274399
https://www.zhihu.com/question/270578639/answer/356272458
https://zhuanlan.zhihu.com/p/36353529
3、caffe2&pytorch模型部署:https://zhuanlan.zhihu.com/p/32342366