首先说明,pytorch最新版本11.8有点问题,安装可以完成,但是用不了cuda,即
import torch
>>> print(torch.cuda.is_available()) #输出为False
所以在这里安装的不是最新版本的pytorch-gpu版本,一般选择安装命令里面有cudatoolkit=x.x的就可以。
1、Linux下安装Anaconda
首先官网下载Anaconda,官网链接Free Download | Anaconda
(1)第一种方法直接wget命令下载
选择合适的linux版本,可以使用wget命令下载:
网址点击下载就会弹出,在windows弹出,复制过去就行

wget https://repo.anaconda.com/archive/Anaconda3-2023.03-1-Linux-x86_64.sh
然后使用bash命令安装
bash Anaconda3-2023.03-1-Linux-x86_64.sh
最后退出用户,重启,就会自动弹出默认的base环境
(2)第二种方法,服务器网差,wget不成功
下载本地,上图直接选择下载就行,然后在本地传进服务器
scp -P 端口号(端口号可选,没有就不写-P 端口号) Anaconda3-2023.03-1-Linux-x86_64.sh username@ip地址:./
传完,在服务器,执行
bash Anaconda3-2023.03-1-Linux-x86_64.sh
最后退出用户,重启,就会自动弹出默认的base环境
2、Linux下安装pytorch-gpu
在pytorch官网Start Locally | PyTorch,选择这里

找到带有cudatoolkit=x.x的pytorch版本就行

复制命令直接安装
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=10.2 -c pytorch
3、直接conda命令安装老是下载失败不成功
(1)换镜像
一般换清华镜像,操作参考这篇博客Conda配置环境常见的几个问题及解决方法_哼唧哼唧啊哈的博客-优快云博客
清华镜像官网anaconda | 镜像站使用帮助 | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror
vim ~/.condarc #打开文件
输入i之后 ,粘贴如下内容
channels:
- defaults
show_channel_urls: true
default_channels:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
pytorch-lts: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
点击ESC,输入:wq,回车,保存退出
再重新安装,就会加速。
(2)使用mamba加速
运行下列代码安装mamba
conda config --append channels conda-forge
conda config --set channel_priority flexible
conda install mamba
安装完成后所有conda可以换成 mamba,比如
mamba install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=10.2 -c pytorch
速度会快很多。
4、安装完成需要测试一下,如下就是成功

文章介绍了在Linux环境下如何安装Anaconda和PyTorch-GPU,特别是针对PyTorch最新版本11.8存在的CUDA不可用问题,提出了选择合适带cudatoolkit的PyTorch版本进行安装的方法。同时,提供了更换镜像源(如清华镜像)和使用mamba加速安装的解决方案。
1331

被折叠的 条评论
为什么被折叠?



