pytorch 安装检测代码

绀目澄清

已于 2023-09-17 21:24:15 修改

阅读量1.2k

点赞数 1

分类专栏： pytorch 文章标签： pytorch 人工智能 python

于 2022-09-24 21:36:45 首次发布

本文链接：https://blog.csdn.net/u013628121/article/details/127031359

版权

PyTorch CUDA Python环境版本检查 GPU支持

关键词由CSDN通过智能技术生成

pytorch 专栏收录该内容

19 篇文章 5 订阅

订阅专栏

import sys

print(sys.path)
print(sys.executable)

import torch

print('torch.__version__ =',torch.__version__)  # pytorch版本
print('torch.version.cuda =',torch.version.cuda)  # cuda版本
print('torch.cuda.is_available =',torch.cuda.is_available())  # 查看cuda是否可用