Pytorch安装

1、pytorch下载

打开网站https://pytorch.org/,会自动根据电脑配置给出下载命令,我的是:

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

2、添加镜像

由于网络问题,直接运行上面命令无法下载,出现http错误的问题,需要添加镜像。
添加镜像后,在c:\Users\用户名 这个目录下生成.condarc文件,即配置的镜像路径。
根据博客https://blog.csdn.net/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位置:
Pytorch安装_第1张图片

参考:
1、pytorch在windows10上安装使用:https://blog.csdn.net/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

你可能感兴趣的:(深度学习与计算机视觉)