Java、Python项目定制找Java1234_小锋老师,专业又靠谱 QQ:3320160706
基于PyTorch-VLM的大模型训练&推理 PDF 下载
时间:2026-02-19 17:11
来源:http://www.java1234.com
作者:转载
侵权举报
基于PyTorch-VLM的大模型训练&推理
| 失效链接处理 |
基于PyTorch-VLM的大模型训练&推理 PDF 下载
相关截图:
主要内容:
3. CUDA+PyTorch镜像
3.1 NVIDIA官方镜像
root@dsw-1581636-75d64b899b-qd972:/mnt/workspace
root@dsw-1581636-75d64b899b-qd972:/mnt/workspace
python -c "
import torch, transformers, platform, distro, os
print(f'Linux版本: {platform.system()} {platform.release()}')
print(f'Ubuntu版本: {distro.name()} {distro.version()}')
print(f'PyTorch版本: {torch.__version__}')
print(f'CUDA版本: {torch.version.cuda}')
print(f'cuDNN版本: {torch.backends.cudnn.version() if
torch.backends.cudnn.is_available() else \"无cuDNN\"}')
print(f'CUDA是否可用: {torch.cuda.is_available()}')
print(f'GPU型号: {torch.cuda.get_device_name(0) if torch.cuda.is_available() else
\"无GPU\"}')
print(f'Transformers版本: {transformers.__version__}')
"
Linux版本: Linux 5.10.134-17.3.al8.x86_64
Ubuntu版本: Ubuntu 22.04
PyTorch版本: 2.3.1+cu121
CUDA版本: 12.1
cuDNN版本: 8902
CUDA是否可用: True
GPU型号: NVIDIA A10
Transformers版本: 4.45.0.dev0
|
------分隔线----------------------------
- 关注Java1234微信公众号
-
锋哥推荐