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.")
相关推荐
yuluo_YX3 分钟前
VSR 项目解析
人工智能·python
cdming40 分钟前
微软Win11双AI功能来袭:“AI管家”+聊天机器人重构桌面交互体验
人工智能·microsoft·机器人
计算衎1 小时前
python通过win32com库调用UDE工具来做开发调试实现自动化源码,以及UDE的知识点介绍
python·c/c++·pywin32·ude·com api
Full Stack Developme1 小时前
java.nio 包详解
java·python·nio
罗西的思考1 小时前
[Agent] ACE(Agentic Context Engineering)和Dynamic Cheatsheet学习笔记
人工智能·机器学习
fantasy_arch1 小时前
transformer-注意力评分函数
人工智能·深度学习·transformer
逐云者1231 小时前
自动驾驶强化学习的价值对齐:奖励函数设计的艺术与科学
人工智能·机器学习·自动驾驶·自动驾驶奖励函数·奖励函数黑客防范·智能驾驶价值对齐
BreezeJuvenile1 小时前
深度学习实验一之图像特征提取和深度学习训练数据标注
人工智能·深度学习
Dev7z2 小时前
舌苔舌象分类图像数据集
人工智能·分类·数据挖掘
新手村领路人2 小时前
opencv gpu cuda python c++版本测试代码
python·opencv·cuda