PyTorch 1.0.0 简易安装流程

笔者在跑一个模型时突然需要用到 PyTorch 1.0.0，以下是笔者使用的各依赖环境版本及下载方式

首先需要安装 Python 和 Anaconda

🗃️附件

Python 3.7.6 下载

https://www.python.org/ftp/python/3.7.6/python-3.7.6-amd64.exe

Anaconda 2020.2 下载

https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-2020.02-Windows-x86_64.exe

⭐推荐

安装过程中具体选项选择可参考教程

https://www.bilibili.com/video/BV1SD4y1S7jo?p=2

在 Anaconda 中打开 Anaconda Prompt

即可根据各 CUDA 版本或仅 CPU 版本输入以下对应的 PyTorch1.0.0 安装代码

# CUDA 10.0
conda install pytorch==1.0.0 torchvision==0.2.1 cuda100 -c pytorch

# CUDA 9.0
conda install pytorch==1.0.0 torchvision==0.2.1 cuda90 -c pytorch

# CUDA 8.0
conda install pytorch==1.0.0 torchvision==0.2.1 cuda80 -c pytorch

# CPU Only
conda install pytorch-cpu==1.0.0 torchvision-cpu==0.2.1 cpuonly -c pytorch

💬相关

更多 PyTorch 版本可翻阅官网

PyTorch 最新版本

https://pytorch.org/get-started/locally/

PyTorch 历史版本

https://pytorch.org/get-started/previous-versions/

安装成功检测代码

import torch
import torchvision
print(torch.__version__)
#显示PyTorch版本
print(torch.cuda.is_available())
#显示True则表明安装PyTorch成功

笔者安装完 PyTorch 后，运行检测代码出现了两个报错，其中一个如下

Undefined symbol: cblas_sgemm_alloc

解决该报错需要在 Anaconda Prompt 中输入以下代码

conda install -c anaconda mkl

另一个报错如下

ImportError: cannot import name 'PILLOW_VERSION' from 'PIL'

原因则是用 conda 安装 PyTorch 的话，默认安装依赖库中的 pillow 库版本较高，但此版本的 pillow 已经移除了 PILLOW_VERSION，因而需要给 pillow 降级，比如说6.1版本

在 Anaconda Prompt 中输入以下代码安装低版本 pillow 库

pip uninstall pillow
pip install pillow==6.1

即可解决报错问题