安装PyTorch过程中遇到的个别小坑

由于众所周知的原因,我们在在线安装PyTorch 1.3或1.4时,会遇到PyTorch主whl包无法下载的状况。这是因为PyTorch的包并没有放在anaconda以及它的free通道和镜像源上,所以会从pytorch官方指定的地址下载包,而这个文件存储在亚马逊S3上,因此下载时几乎无法下载。因此我们通过一点特殊方法下载到whl文件,我们从

https://download.pytorch.org/whl/torch_stable.html

找到我们要下载的版本,比如是Python 3.7的还是3.6的,是CPU还是GPU版的。我们要下载两个包,一个是torch一个是torchvision

首先安装Python。步骤如下:

创建名为pytorchlab的环境并安装python 3.7

conda create -n pytorchlab python=3.7

环境创建完毕后激活环境

conda activate pytorchlab

而后继续安装whl和相关必要的包

pip install pypiwin32,pillow==6.1

安装win32api。安装jupyter时遇到了一个问题,好像没有下载win32api支持模块。我也不知道为什么。注意需要安装pillow 6.1版本,自动安装的7.0版本会导致一个变量错误

安装whl文件。

GPU版
 

pip install torch-1.4.0-cp37-cp37m-win_amd64.whl

pip install torchvision-0.4.2-cp37-cp37m-win_amd64.whl

CPU版
 

pip install torch-1.4.0+cpu-cp37-cp37m-win_amd64

pip install torchvision-0.4.2+cpu-cp37-cp37m-win_amd64.whl

最后安装jupyter 和jupyterlab

conda install jupyter jupyterlab 

最后在你的工作目录中启动jupyter notebook 或 lab即可。

你可能感兴趣的:(机器学习,pytorch)