在虚拟环境中，安装gpu版本的torch

虚拟环境中，安装gpu版本的torch

1. 下载离线包

根据所需要的版本下载相应的离线包
离线包下载地址：https://download.pytorch.org/whl/torch_stable.html

（1）下载torch

• cu118 表示 cuda11.8
• cpu 表示 cpu版本
• cp39 表示 python 3.9 版本
• win和linux 表示操作系统

（2）同理，可以下载torchvison,torchaudio等

2. 安装

进入虚拟环境，cd到这些whl文件存储的目录
使用指令安装

pip install XXXXXXX.whl

可以使用tab键补全whl文件名

3. 查看是否完成安装

（1）方法一

在相应的虚拟环境中，使用 pip list ,查看完成安装的第三方库

带cu的 即为cuda版本

（2）方法二

import torch
print(torch.__version__)

查看torch是否为cuda版本

（3）方法三

print(torch.cuda.is_available())

True 即 安装成功