由于老唐之前都是使用同学装好的 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 和在线阅读》。