加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
1.py 339 Bytes
一键复制 编辑 原始数据 按行查看 历史
zcl 提交于 2022-06-10 20:23 . first
# -*- coding:utf-8 -*-
# 作者:19645 日期:2022年06月08日
import torch
print(torch.__version__) # 查看torch当前版本号
print(torch.version.cuda) # 编译当前版本的torch使用的cuda版本号
print(torch.cuda.is_available()) # 查看当前cuda是否可用于当前版本的Torch,如果输出True,则表示可用
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化