一、Terminal输出命令:nvidia -smi
查看结果:
得到 CUDA Version为12.3
然后去torch官网查看CUDA对应的torch版本
复制命令在Anaconda环境中运行
再Python解释器中使用该虚拟环境并复制以下代码:
import torch
print("是否可用:", torch.cuda.is_available()) # 查看GPU是否可用
print("GPU数量:", torch.cuda.device_count()) # 查看GPU数量
print("torch方法查看CUDA版本:", torch.version.cuda) # torch方法查看CUDA版本
print("GPU索引号:", torch.cuda.current_device()) # 查看GPU索引号
运行如下:
注意:若不成功则显示False