Manjaro linux(KDE)安装pytorch(cuda)小记&方法参考

评价:不比windows下安装复杂

首先弄明白:anaconda安装pytorch是会自带指定版本的cuda toolkit的,不需要手动安装。这种安装方式的依赖只有nvidia的驱动,而且Manjaro在系统安装时就可以指定安装私有版也就是闭源的nvidia显卡驱动了。
Manjaro linux(KDE)安装pytorch(cuda)小记&方法参考_第1张图片
因此,整个安装流程就和Manjaro的界面一样丝般顺滑

安装流程

1.首先下载Manjaro(KDE版)的ISO镜像,我这里选择了官网提供的种子下载的https://manjaro.org/get-manjaro/
2.u盘烧录我这里用了Etcher,没问题。
3.注意: 在安装u盘boot的时候要选择driver=nonfree,如果和我一样是轻薄本(1145g7)+外接nvidia GPU的话要插上显卡。

接下来正常系统安装,开机。

4.不出意外的话打开系统设置->硬件设定->驱动->Display controller中能看到你的GPU和至少一个私有的驱动。
终端输入nvidia-smi
Manjaro linux(KDE)安装pytorch(cuda)小记&方法参考_第2张图片
看得到GPU和驱动信息的话就可以愉快下一步了。
6.anaconda安装:这里选择了清华开源软件镜像站下载https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
找到最新的anaconda(linux-x86_64)的最新版本即可。
终端cd到下载目录,命令

 bash Anaconda3-....sh

可以看到~/anaconda/bin里都是能执行的东西,需要把这个文件夹设置好环境变量:

nano ~/.bashrc

在这个文件的最后一行加入:(例如)

export PATH="/home/insppp/anaconda3/bin:$PATH"

路径是你刚刚看到的bin文件夹的绝对路径
顺手进行Anaconda 的换源:
将~/.condarc用同样的方式修改为如下内容:

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

7.conda环境安装pytorch‘
新建环境

conda create --name mytorch python=3.7

进入环境

conda activate mytorch

在pytorch官网找到你想要的安装命令https://pytorch.org/get-started/locally/
Manjaro linux(KDE)安装pytorch(cuda)小记&方法参考_第3张图片
比如我的就是

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

这里会直接安装一个对应版本的cuda toolkit供pytorch调用,不用担心系统的cuda版本
8.验证安装
此时再nvidia-smi或者别的查看看到的cuda版本可能是和pytorch的不相同的,没关系,验证如下:
Manjaro linux(KDE)安装pytorch(cuda)小记&方法参考_第4张图片
Manjaro linux(KDE)安装pytorch(cuda)小记&方法参考_第5张图片只要这几条python正常执行就可以了。

你可能感兴趣的:(pytorch,linux,深度学习,manjaro,python)