安装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即可。
上一篇: 冒泡排序(倒序)