文章目录
展开
由于老唐之前都是使用同学装好的 GPU 服务器的，一直没有自己装过，今天自己装了一下，发现还挺复杂的。本文先记录一下如何使用 PyTorch 测试 CUDA 是否安装成功，测试 CUDA 是否安装成功还是挺简单的，由于一直使用 PyTorch 比较多，所以本文就分享通过 PyTorch 进行测试的方法，使用 TensorFlow 测试的方法应该大同小异。
一、测试 CUDA 是否安装成功
很简单，只要下面两行代码：
import torch print(torch.cuda.is_available())
返回结果是 True，则 PyTorch 的 GPU 安装成功；否则就是没有成功。
二、参考文献和其他
参考文献：https://blog.csdn.net/qq_38410428/article/details/82965499
老唐整理的其他 PyTorch 教程：《PyTorch 使用指定的 GPU 的方法》。