Python 查看pytorch版本、判断CUDA是否可用

查无此人☞

129786人浏览 · 2021-03-15 21:00:52

查无此人☞ · 2021-03-15 21:00:52 发布

在深度学习跑论文代码的时候，安装好环境后，经常会验证torch的版本、以及torch与cuda版本是否对应、cuda是否可用、以及torch对应的cuda的版本。代码如下！

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

结果如图:
在这里插入图片描述
也可查看可行的cuda数目：

print(torch.cuda.device_count())

在这里插入图片描述
查看torch对应的cuda版本：

torch.version.cuda

在这里插入图片描述

脑启社区

脑启社区是一个专注类脑智能领域的开发者社区。欢迎加入社区，共建类脑智能生态。社区为开发者提供了丰富的开源类脑工具软件、类脑算法模型及数据集、类脑知识库、类脑技术培训课程以及类脑应用案例等资源。

更多推荐

具身智能VLA模型全景图：从Pi0到HoloBrain-0的15大开源力作

从Physical Intelligence的Pi0到地平线的HoloBrain-0，这15个模型不仅代表了当前VLA技术的水平，更展示了具身智能从‘单一任务’向‘通用操作’迈进的坚定步伐。无论是追求极致的灵巧控制，还是探索类脑架构的安全性，开源社区正在以前所未有的速度缩小仿真与现实的差距。同步开源了仅 0.2B 参数的端侧友好版本及全栈基础设施 RoboOrchard，提供从数据采集、标准化规范