一、检测脚本
import torch
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.version.cuda)
输出
二、解决排查
1、安装CUDA
根据驱动选择要安装的版本
安装网站:CUDA Toolkit Archive | NVIDIA Developer
安装成功后
2、安装cuDNN
注意:需要注册才能进行安装
具体
一、检测脚本
import torch
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.version.cuda)
输出
二、解决排查
1、安装CUDA
根据驱动选择要安装的版本
安装网站:CUDA Toolkit Archive | NVIDIA Developer
安装成功后
2、安装cuDNN
注意:需要注册才能进行安装
具体