Pytorch-GPU版本离线安装

最近在复现一项深度学习的工作,发现自己的pytorch是装的cpu版的(好像当时是直接加清华源,默认是cpu版本)。从官网在线下载速度太慢,还时不时断开连接,我们可以配置conda的清华源去这个问题,但是考虑到是在用组内服务器,配崩了可能影响其它人(虽然我本机以前配置过),这次换个思路解决,从清华镜像源下载对应的安装包手动安装。

Pytorch安装命令

shell 复制代码
pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu118
shell 复制代码
nvidia-smi

自己查显卡能支持的版本,我的cuda版本最高支持12.3,但我习惯用cuda11.8

清华镜像源下载对应的pytorch安装包

下载地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/
注意这是linux版本的,windowx下载把linux-64替换成win-64

pytorch+cuda11.8


注意自己的python的版本

torchvision-0.16


注意自己的python的版本

安装pytorch

切换到对应的conda环境

shell 复制代码
conda activate env-name
conda install --offline  /path/to/pytorch-2.1.0-py3.8_cuda11.8_cudnn8.7.0_0.tar.bz2
conda install --offline  /path/to/pytorch-2.1.0-py3.8_cuda11.8_cudnn8.7.0_0.tar.bz2

检查pytorch是否安装成功

python 复制代码
import torch

# 检查是否可以使用 GPU
if torch.cuda.is_available():
    print("CUDA is available! PyTorch can use GPU.")
    print("CUDA version:", torch.version.cuda)
    print("Number of GPUs available:", torch.cuda.device_count())
    print("Current device name:", torch.cuda.get_device_name(0))
else:
    print("CUDA is not available. PyTorch cannot use GPU.")
相关推荐
txwtech13 小时前
第9篇 opencv提取矩形角度不是很准确的处理方法
人工智能·opencv·计算机视觉
却道天凉_好个秋13 小时前
OpenCV(十二):Mat
人工智能·opencv·计算机视觉
TreeExplore13 小时前
OpenCV工程中直接包含调用vs2022
人工智能·opencv·计算机视觉
极客数模14 小时前
2025年MathorCup 大数据竞赛明日开赛,注意事项!论文提交规范、模板、承诺书正确使用!2025年第六届MathorCup数学应用挑战赛——大数据竞赛
大数据·python·算法·matlab·图论·比赛推荐
Juchecar14 小时前
从瑟瑟发抖到电流“高速公路”:揭秘能量与神奇的微观世界
人工智能
PieroPc14 小时前
用Python Streamlit Sqlite3 写一个简单商品管理系统
数据库·python·sqlite·streamlit
小白银子14 小时前
零基础从头教学Linux(Day 55)
java·linux·服务器·python
珊珊而川14 小时前
vllm论文中figure3每个块的区别
人工智能
程序员小远14 小时前
Appium+python+unittest搭建UI自动化框架
自动化测试·软件测试·python·测试工具·ui·appium·测试用例
北海道浪子14 小时前
[免费送$1000]ClaudeCode、Codex等AI模型在开发中的使用
前端·人工智能·后端