admin 管理员组

文章数量: 1184232

最开始,创建环境,用这个代码查看显卡

import torch
print(torch.cuda.is_available())
print(torch.version.cuda)
print(torch.cuda.get_device_name(0))

返回都是false

修改方法:

用这个跑在创建好的环境里

conda install pytorch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 pytorch-cuda=12.1 -c pytorch -c nvidia 

再跑第一个代码,就能找到

本文标签: 轻松掌握 编程 电脑硬件