ubuntu下anconda装pytorch

li三河2024-12-17 10:33

1、禁用nouveau

复制代码

sudo vim /etc/modprobe.d/blacklist.conf

在文件最后部分插入以下两行内容

复制代码

blacklist nouveau
options nouveau modeset=0

更新系统

复制代码

sudo update-initramfs -u

重启系统

2、装nvidia驱动

卸载原来驱动

复制代码

  sudo apt-get remove nvidia-*  （若安装过其他版本或其他方式安装过驱动执行此项）

给驱动run文件赋予执行权限：

复制代码

sudo chmod  a+x NVIDIA-Linux-x86_64-418.43.run

安装

复制代码

sudo ./NVIDIA-Linux-x86_64-418.43.run -no-x-check -no-nouveau-check -no-opengl-files

3、装anconda

清华源下载符合自己系统的最新版（尽可能下载新的，否则pytorch也不能安装比较新的版本）Index of /https://repo.anaconda.com/archive/配置安装清华源：

复制代码

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
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/r
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
conda config --set show_channel_urls yes

3、装pytorch

复制代码

conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple xxx