Torch GPU版本的安装

*不需要单独安装巨大的CUDA安装包， 先确保你的显卡是支持GPU运算的，

再安装好anaconda
确保没有安装：pytorch torchvision torchaudio这三个模块
然后在anaconda中的命令行输入：

conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia -y

其中12.4是你要安装CUDA的版本，可跟根需要修改。

等待漫长的在线下载安装过程即可（如果没有KX上网的话，可能需要数个小时甚至更长）

安装完成后，可用下面代码验证：

`

import torch
import torchvision
print("PyTorch版本: ", torch.version) # 打印PyTorch版本
print("torchvision版本 ", torchvision.version) # 打印torchvision版本
print("CUDA是否可用: ", torch.cuda.is_available()) # 检查CUDA是否可用

返回True,就说明CUDA激活成功。