笔者在跑一个模型时突然需要用到 PyTorch 1.0.0,以下是笔者使用的各依赖环境版本及下载方式
首先需要安装 Python 和 Anaconda
????️附件
Python 3.7.6 下载
/ftp/python/3.7.6/python-3.7.
Anaconda 2020.2 下载
/anaconda/archive/Anaconda3-2020.02-Windows-x86_64.exe
⭐推荐
安装过程中具体选项选择可参考教程
/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 最新版本
/get-started/locally/
PyTorch 历史版本
/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
即可解决报错问题