PyTorch 安装

PyTorch 安装

---

1、安装

官网： https://pytorch.org/

conda install pytorch -c pytorch
pip3 install torchvision

根据自己的需求选择要安装的版本，在anaconda上执行提示的命令。

---

2、测试是否成功安装

# 导入
import torch

# pytorch 的版本
torch.__version__

# cuda 是否可用
torch.cuda.is_available()   >>>True

---

3、可能遇到的问题

官网解答： https://pytorch.org/docs/stable/notes/windows.html

问题1： Import error

# import torch时出现以下错误：
from torch._C import *
ImportError: DLL load failed: The specified module could not be found.

# 解决方案：缺少了必要的一些动态链接库，或者没英伟达显卡而误装了GPU版本的
conda install -c peterjc123 vc vs2017_runtim
conda install mkl_fft intel_openmp numpy mkl

---