赞
踩
参考:
https://blog.csdn.net/m0_46948660/article/details/129205116
报错处理:
进入环境:
conda activate xxx
安装torch:
conda install torch
引入及测试:
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索引号
print("GPU名称:", torch.cuda.get_device_name(1)) # 根据索引号得到GPU名称
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。