Linux 服务器下Pytorch安装（版本对应）--清华源

 版本对应

python=3.9.19
pytorch=2.4.0

基于cuda驱动11.8

安装流程

安装anaconda–>python下载–>pytorch下载

以前的文章年限较久。对于新安装的朋友给出新版本对应关系，也做个人纪录。

安装anaconda

安装好后。conda下搭建新的python环境

conda create -n yolov5_ python=3.9.19

yolov5_可替换为个人的环境名称

#激活环境
conda activate yolov5_

添加清华源，必要（直接打到命令行）

# 添加清华源的pytorch
conda config –add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config –add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config –set show_channel_urls yes
conda config –add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

查看是否修改正确

conda config –show channels

开始下载pytorch，不需要删除-c pytorch 等。此时镜像源已经修改好了

conda install pytorch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 pytorch-cuda=11.8 -c pytorch -c nvidia

完成后检查是否成功

Python 3.9.19 (main, May 6 2024, 19:43:03)
[GCC 11.2.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> print(torch.cuda.is_available())
True
>>> print(torch.__version__)
2.4.0
>>> torch.version.cuda
'11.8'
>>>