使用 PyTorch 测试 CUDA 是否安装成功

已关闭留言

由于老唐之前都是使用同学装好的 GPU 服务器的,一直没有自己装过,今天自己装了一下,发现还挺复杂的。本文先记录一下如何使用 PyTorch 测试 CUDA 是否安装成功,测试 CUDA 是否安装成功还是挺简单的,由于一直使用 PyTorch 比较多,所以本文就分享通过 PyTorch 进行测试的方法,使用 TensorFlow 测试的方法应该大同小异。

文章目录
隐藏
一、测试 CUDA 是否安装成功
二、参考文献和其他

一、测试 CUDA 是否安装成功

很简单,只要下面两行代码:

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

返回结果是 True,则 PyTorch 的 GPU 安装成功;否则就是没有成功。

二、参考文献和其他

参考文献:https://blog.csdn.net/qq_38410428/article/details/82965499

老唐整理的其他 PyTorch 教程:《PyTorch 使用指定的 GPU 的方法》。

机器学习教程:《李宏毅机器学习笔记 (LeeML-Notes) 最全整理 GitHub 和在线阅读》。