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激活成功。

posted @   Cooker2008  阅读(129)  评论(0编辑  收藏  举报
努力加载评论中...
点击右上角即可分享
微信分享提示