在安装pytorch的过程中,尝试过在线安装,离线安装清华网站镜像资源,pytorch官方网站离线安装等等方式后,认为最简单并且最不容易出错的安装方式如下:
首先,前提是已经安装好合适版本的cuda、python和anaconda
进入anaconda prompt
- 创建环境 conda create -n pytorch-gpu python=3.6(我用的3.6,写你自己python的版本号)
- pytorch官方网站下载torch和torchvision的离线whl安装包,torch和torchvision的版本一定要对应,我下的是torch1.5.0和torchvision0.6.0,cuda版本10.1(尝试过从清华镜像网站上下载压缩包资源,但安装后一直各种出错,无法使用) 离线安装包下载地址: https://download.pytorch.org/whl/torch_stable.html
图中torch后面的数字代表torch版本,cu101代表cuda10.1,cp36代表python3.6版本,win代表windows系统,torchvision同理,一定要对应自己所需要的版本!
torch与torchvision对应版本网址:https://pytorch.org/get-started/previous-versions/
- 激活刚刚创建的环境 activate pytorch-gpu
- 激活后安装已经下载好的安装包 pip install 压缩包保存路径\压缩包名
- 查看已经安装的压缩包 pip list
- 输入python
- 输入 import torch,没有报错即安装成功
- 输入 torch.cuda.is_available()查看是否能调用gpu