测试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
相关推荐
databook10 小时前
Manim实现闪光轨迹特效
后端·python·动效
Juchecar11 小时前
解惑:NumPy 中 ndarray.ndim 到底是什么?
python
用户83562907805111 小时前
Python 删除 Excel 工作表中的空白行列
后端·python
Json_11 小时前
使用python-fastApi框架开发一个学校宿舍管理系统-前后端分离项目
后端·python·fastapi
CoovallyAIHub13 小时前
中科大DSAI Lab团队多篇论文入选ICCV 2025,推动三维视觉与泛化感知技术突破
深度学习·算法·计算机视觉
CoovallyAIHub13 小时前
开源的消逝与新生:从 TensorFlow 的落幕到开源生态的蜕变
pytorch·深度学习·llm
数据智能老司机18 小时前
精通 Python 设计模式——分布式系统模式
python·设计模式·架构
数据智能老司机19 小时前
精通 Python 设计模式——并发与异步模式
python·设计模式·编程语言
数据智能老司机19 小时前
精通 Python 设计模式——测试模式
python·设计模式·架构
数据智能老司机19 小时前
精通 Python 设计模式——性能模式
python·设计模式·架构