pytorch安装

一、环境说明

• 语言：Python3.7
• 系统：win10

在家里重新配置环境，忘记了上次是怎么配置的，经过一番折腾后，还是决定记下来

二、安装步骤

• 添加清华源¹

国外源奇慢无比，还容易中断。使用国内清华镜像源

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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

• 安装pytorch²

打开cmd，输入以下指令，如果找不到conda这个指令，就使用Anaconda自带的 Anaconda Powershell Prompt的工具输入以下指令

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

• 出现的问题

如下图所示，pytorch-1.1.0依旧无法从清华源获取，只能使用默认的国外源，网络经常中断。

所以pytorch-1.1.0毫无意外的安装到一半失败了

此时决定亲自去一趟清华源网站³看看，进入anaconda目录，再进入cloud目录，再进入pytorch目录，再选择对应版本，根据上述图片所需要的下载的库py3.7_cuda90_cudnn7_1，全局查找。下载到本地（推荐使用idm下载器下载），后使用以下命令进行安装。

conda install --use-local D:\Users\22269\Downloads\Compressed\pytorch-1.1.0-py3.7_cuda90_cudnn7_1.tar.bz2

安装成功

• 2019年12月15号更新：

使用以下命令可以直接安装，-c pytorch指的是从conda官方下载，当然是从清华源下载好，所以改成下面指令，可以从清华源直接安装：

conda install pytorch torchvision cudatoolkit=9.0

三、安装OpenCV

输入以下命令：

conda install -c menpo opencv

所有依赖库都会自动安装，可以看到所有的都找到了国内源，下载速度飞快

全部下载成功

四、检测

经过检测，两个库都成功安装

但是最后不知道为什么，import torch的时候，表示无法找到对应dll文件，然后怀疑可能是手动安装的pytorch有问题，于是再次执行一次以下命令：

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

完美运行。

猜测可能手动安装缺少最后一个执行动作

参考文献

¹. 欢呼、雀跃，Anaconda 清华源又回来了！ ↩

². Pytorch官网 ↩

³. 清华大学开源软件镜像站 ↩