一、查看本机能安装的cuda版本
1、已安装
nvcc --version
这里可以看出cuda版本为11.3
2、未安装
nvidia-smi
显示的是最大课安装cuda版本
二、下载对应的torch组件
1、先查看自己的python环境
python -V
2、找到对应python环境下的torch组件版本
方法一、
打开官网:Previous PyTorch Versions | PyTorch
ctrl+F搜索 torch==x.xx.x查看官网给的版本
方法二、
百度
3、下载.whl文件
使用离线下载的方式,下载网址:download.pytorch.org/whl/cu102
分别下载torch、torchaudio、torchvision
三、创建环境
1、创建环境、激活环境、进入.whl文件夹下
conda create -n your_envs_name python=x.x
conda activate your_envs_name
cd 进入.whl文件夹
2、使用pip install安装
pip install torch-1.10.2+cu113-cp37-cp37m-win_amd64.whl
pip install torchaudio-0.10.2+cu113-cp37-cp37m-win_amd64.whl
pip install torchvision-0.11.3+cu113-cp37-cp37m-win_amd64.whl
四、查看是否安装正确
进入python环境运行
python
import torch
torch.cuda.is_available()
显示ture即为安装正确