在安装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
![]()
- **刚刚创建的环境 activate pytorch-gpu
- **后安装已经下载好的安装包 pip install 压缩包保存路径\压缩包名
![]()
- 查看已经安装的压缩包 pip list
- 输入python
- 输入 import torch,没有报错即安装成功
- 输入 torch.cuda.is_available()查看是否能调用gpu
![]()