测试pytorch(GPU)安装,并输出版本号

目录

GPU版本pytorch安装测试

CUDA信息显示

GPU版本pytorch安装测试

我们一般完成pytorch(GPU)版本后,要进行测试,以下代码用于备注:

python 复制代码
import torch
flag = torch.cuda.is_available()
print(flag)

ngpu= 1
device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
print(device)
print(torch.__version__)
print(torch.cuda.get_device_name(0))
print(torch.rand(3,3).cuda())

测试结果如下:

True

cuda:0

2.0.0+cu118

NVIDIA GeForce RTX 3090

tensor([[0.0756, 0.7062, 0.6051],

0.2772, 0.5758, 0.3481\], \[0.4451, 0.9666, 0.9337\]\], device='cuda:0')

CUDA信息显示

在CUDA后续的更新中,windows系统的任务管理器看不见CUDA占用了,在Ubuntu中同样无法看见,需要在命令行窗口输入:

bash 复制代码
nvidia-smi
相关推荐
站大爷IP1 小时前
Java调用Python的5种实用方案:从简单到进阶的全场景解析
python
用户8356290780517 小时前
从手动编辑到代码生成:Python 助你高效创建 Word 文档
后端·python
c8i7 小时前
python中类的基本结构、特殊属性于MRO理解
python
隐语SecretFlow7 小时前
国人自研开源隐私计算框架SecretFlow,深度拆解框架及使用【开发者必看】
深度学习
liwulin05067 小时前
【ESP32-CAM】HELLO WORLD
python
Doris_20238 小时前
Python条件判断语句 if、elif 、else
前端·后端·python
Doris_20238 小时前
Python 模式匹配match case
前端·后端·python
Billy_Zuo8 小时前
人工智能深度学习——卷积神经网络(CNN)
人工智能·深度学习·cnn
这里有鱼汤8 小时前
Python量化实盘踩坑指南:分钟K线没处理好,小心直接亏钱!
后端·python·程序员
羊羊小栈8 小时前
基于「YOLO目标检测 + 多模态AI分析」的遥感影像目标检测分析系统(vue+flask+数据集+模型训练)
人工智能·深度学习·yolo·目标检测·毕业设计·大作业