打开终端,切换到pytorch安装环境下,检测程序如下:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
输出结果:
表示GPU安装成功,pytorch版本为1.3.0
打开终端,切换到pytorch安装环境下,检测程序如下:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
输出结果:
表示GPU安装成功,pytorch版本为1.3.0