pycharm 中使用pytorch GPU

pycharm 中使用pytorch GPU

安装版本对应的cuda，cudnn，torch

我安装的：
cuda：11.6
cudnn：8.4
pytorch：1.14

安装torch命令：

pip install torch torchvision torchaudio --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu116

检查cuda是否可用：

if_cuda = torch.cuda.is_available()
print("if_cuda=",if_cuda)

import torch
print("Pytorch version：")
print(torch.__version__)
print("CUDA Version: ")
print(torch.version.cuda)
print("cuDNN version is :")
print(torch.backends.cudnn.version())

print("torch.cuda.is_available():",torch.cuda.is_available())           # cuda是否可用
print("torch.cuda.current_device():",torch.cuda.current_device())       # 返回当前设备索引
print("torch.cuda.device_count():",torch.cuda.device_count())           # 返回GPU的数量
print("torch.cuda.get_device_name(0):",torch.cuda.get_device_name(0))   # 返回gpu名字，设备索引默认从0开始