JustHost 华纳云 搬瓦工 GigsGigsCloud DMIT 腾讯云

使用 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 和在线阅读》。

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