搬瓦工 GigsGigsCloud 腾讯云

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

搬瓦工

由于老唐之前都是使用同学装好的 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 的方法》。

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

赞(1)
版权声明:本文采用知识共享 署名4.0国际许可协议 [BY-NC-SA] 进行授权
文章名称:《使用 PyTorch 测试 CUDA 是否安装成功
文章链接:https://oldtang.com/827.html
本站资源仅供个人学习交流,请于下载后24小时内删除,不允许用于商业用途,否则法律问题自行承担。