Java知识分享网 - 轻松学习从此开始!    

Java知识分享网

        
AI编程,程序员挑战年入30~100万高级指南 - 职业规划
SpringBoot+SpringSecurity+Vue权限系统高级实战课程        

IDEA永久激活

Java微信小程序电商实战课程(SpringBoot+VUe)

     

AI人工智能学习大礼包

     

PyCharm永久激活

66套java实战课程无套路领取

     

Cursor+Claude AI编程 1天快速上手视频教程

     
当前位置: 主页 > Java文档 > 人工智能AI >

基于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# pull registry.dockercn.com/nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04
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



 

 
------分隔线----------------------------


锋哥推荐