更加适合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")
相关推荐
舒一笑23 分钟前
智能体革命:企业如何构建自主决策的AI代理?
人工智能
SpikeKing1 小时前
Server - 使用 Docker 配置 PyTorch 研发环境
pytorch·docker·llm
丁先生qaq1 小时前
热成像实例分割电力设备数据集(3类,838张)
人工智能·计算机视觉·目标跟踪·数据集
Eiceblue1 小时前
Python读取PDF:文本、图片与文档属性
数据库·python·pdf
红衣小蛇妖1 小时前
神经网络-Day45
人工智能·深度学习·神经网络
weixin_527550402 小时前
初级程序员入门指南
javascript·python·算法
KKKlucifer2 小时前
当AI遇上防火墙:新一代智能安全解决方案全景解析
人工智能
程序员的世界你不懂2 小时前
Appium+python自动化(十)- 元素定位
python·appium·自动化
DisonTangor2 小时前
【小红书拥抱开源】小红书开源大规模混合专家模型——dots.llm1
人工智能·计算机视觉·开源·aigc
CryptoPP3 小时前
使用WebSocket实时获取印度股票数据源(无调用次数限制)实战
后端·python·websocket·网络协议·区块链