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

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

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

Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐