更加适合CUDA11.2的pytorch显卡版本

更加适合CUDA11.2的pytorch显卡版本

pip版本

复制代码
pip install torch==1.12.0+cu113 torchvision==0.13.0+cu113 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu113

conda版本

复制代码
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.3 -c pytorch

测试代码

python 复制代码
import torch

if torch.cuda.is_available():
    device = torch.cuda.get_device_name(0)
    print(f"当前GPU型号: {device}")
else:
    print("未检测到可用的GPU")
相关推荐
虹科数字化与AR32 分钟前
安宝特方案丨船舶智造的“AR+AI+作业标准化管理解决方案”(装配)
人工智能·ar·ar眼镜·船舶智造·数字工作流·智能装配
liuzhenghua662 小时前
Python任务调度模型
java·运维·python
飞哥数智坊2 小时前
Coze实战第13讲:飞书多维表格读取+豆包生图模型,轻松批量生成短剧封面
人工智能
小前端大牛马2 小时前
java教程笔记(十一)-泛型
java·笔记·python
sjtu_cjs2 小时前
Tensorrt python api 10.11.0笔记
开发语言·笔记·python
哆啦A梦的口袋呀2 小时前
深入理解系统:UML类图
开发语言·python·uml
虎冯河2 小时前
怎么让Comfyui导出的图像不包含工作流信息,
开发语言·python
newxtc2 小时前
【配置 YOLOX 用于按目录分类的图片数据集】
人工智能·目标跟踪·分类