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

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

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

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐