欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2022-03-21 21:16:44
...

由于众所周知的原因,我们在在线安装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

安装win32api。安装jupyter时遇到了一个问题,好像没有下载win32api支持模块。我也不知道为什么。

安装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即可。