乌班图安装Pytorch Cuda环境

2022-05-19 12:48:15 浏览数 (1)

安装Anaconda

在网址https://www.anaconda.com/products/distribution下载安装包,我这里下载的是Anaconda3-2022.05-Linux-x86_64.sh,它对应的Python版本为3.9。

执行

代码语言:javascript复制
bash Anaconda3-2022.05-Linux-x86_64.sh

安装VIM

代码语言:javascript复制
sudo apt-get install vim

配置Anaconda环境变量

代码语言:javascript复制
sudo vim /etc/profile

添加环境配置(此处的user需要替换成你自己的用户名)

代码语言:javascript复制
export PATH=/home/user/anaconda3/bin:$PATH

使配置可用

代码语言:javascript复制
source /etc/profile

创建Anaconda虚拟环境

代码语言:javascript复制
conda create -n py39 python=3.9.0
source activate
source deactivate
conda activate py39

下载英伟达驱动

网址https://www.nvidia.cn/geforce/drivers/

因为我这里是Geforce RTX 3090,所以在下载页面依次填写

搜索后选择一个驱动

我这里下载的文件名为NVIDIA-Linux-x86_64-470.94.run

执行

代码语言:javascript复制
sudo vim /etc/modprobe.d/blacklist.conf

在末尾添加

代码语言:javascript复制
blacklist nouveau

保存后继续执行

代码语言:javascript复制
sudo update-initramfs -u
reboot

重启后重新激活py39

代码语言:javascript复制
conda activate py39

卸载旧版本的驱动

代码语言:javascript复制
sudo apt-get purge nvidia*
sudo apt-get autoremove

给下载的run文件赋予可执行权限

代码语言:javascript复制
sudo chmod 777 NVIDIA-Linux-x86_64-510.73.05.run

0 人点赞