打开终端,切换到pytorch安装环境下,检测程序如下:

import torch
print(torch.__version__)
print(torch.cuda.is_available())

输出结果:
在这里插入图片描述
表示GPU安装成功,pytorch版本为1.3.0

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐