环境配置
PyTorch的环境配置相对简单,在官网http://pytorch.org中选择相应的操作系统,包管理器,Python版本和CUDA版本(该模块用于GPU加速,如果有支持CUDA的显卡选择一个版本即可),然后进行安装。
笔者基于Anaconda包管理器下的安装指令为
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
此时在windows系统下于Python中执行
import torch
时可能会遇到 'Intel MKL FATAL ERROR: Cannot load mkl_intel_thread.dll.' 的报错,笔者的解决方法是将Anaconda/Library/bin 文件夹中mkl开头的dll文件与libiomp5md.dll复制至Anaconda根目录下。该问题广泛存在,似乎是numpy库的历史遗留问题,此处不做深究。