ubuntu下anconda装pytorch

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
相关推荐
vvw&6 分钟前
如何在 Ubuntu 22.04 上安装 Strapi CMS
linux·运维·服务器·ubuntu·npm·开源·node.js
生椰拿铁You6 分钟前
PHPstudy中的数据库启动不了
linux·运维·服务器
一名陈序员1 小时前
Go-FastDFS文件服务器一镜到底使用Docker安装
linux·服务器·docker
alden_ygq1 小时前
etcd详解
linux·redis·etcd
基哥的奋斗历程2 小时前
Linux 安装 与 Docker 安装 配置镜像
linux·运维·docker
m0_541034842 小时前
Linux shell脚本练习(五)
linux
乌漆嘎嘎黑2 小时前
训练的Loss和输出突然全是nan【小白找bug】
pytorch·python·bug·llama·大模型权重文件
四口鲸鱼爱吃盐2 小时前
Pytorch | 从零构建AlexNet对CIFAR10进行分类
人工智能·pytorch·分类
工业3D_大熊3 小时前
3D可视化引擎HOOPS Visualize与HOOPS Luminate Bridge的功能与应用
linux·前端·c++·windows·macos·3d·c#