使用pycharm调用CUDA

首先在pycharm中使用device是否具有gpu

device = torch.device(args.device if torch.cuda.is_available() else "cpu")
print(device)

运行结果是 cpu ，计算机安装了gpu的，结果没有调用加速。
查了相关的资料 如何在Pycharm中调用gpu 。记录下当时的过程。

在设备管理器中查看自己的gpu

在此电脑—->管理—–>设备管理器中—->显示适配器查看有没有gpu,如图，

下载显卡驱动

在NVIDIA官网中后按照自己电脑的配置选择驱动。

点击搜索，跳转到下载界面

下载安装，安装步骤我选择的是默认，一直到最后。
安装完毕，按下win+R组合键，打开cmd命令窗口。输入如下的命令。

nvidia-smi

会有如下：

我的pycharm是已经安好了，在pytorch官网复制链接，粘贴到pycharm中

到这里，运行下代码

import torch
print(torch.cuda.device_count())
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())
print(torch.cuda_version)
print(torch.backends.cudnn.version())

运行结果：

来源：龙庭花雨落