报错出现源包的问题
解决: 在下包前加上这句
conda config --add channels conda-forge
后续成功安装
import torch
print(torch.cuda.is_available()) # cuda是否可用
print(torch.cuda.current_device()) # 返回当前设备索引
print(torch.cuda.device_count()) # 返回GPU的数量
print(torch.cuda.get_device_name(0))