检查cuda版本
命令行输入
nvidia-smi
安装anaconda3
无过程
官网最新版本下载
https://pytorch.org/get-started/locally/#windows-prerequisites
在conda中创建环境,输入即可
指定版本下载
https://download.pytorch.org/whl/torch_stable.html
找到自己想要的
提示:CTRL+F可以用于页面搜索
下载完成后,打开conda,激活对应的环境
找到下载位置,但注意下,需要先退回根目录才能进去其他盘
安装指令
注意文件名
pip install torch-1.8.1+cu101-cp37-cp37m-win_amd64.whl
检查
检查代码
conda activate 环境
python
import torch
print(torch.__version__)
以上完成torch安装
torchvision安装
下面为版本对应
找到对应版本后,还是在上述的网址上查找
https://download.pytorch.org/whl/torch_stable.html
步骤同上,找对应的包->下载whl->conda激活环境->进入下载地址->pip安装->检查