Ubantu安装后更新驱动、安装Anaconda和Pytorch
本文只是简单介绍一下步骤,方法日新月异,具体方法可以搜其他文章
一、安装Ubantu系统
二、为Ubantu更换源
软件和更新里面-Ubuntu软件(下载自:选择阿里源),如图
三、添加PPA软件仓库:
sudo add-apt-repository ppa:graphics-drivers/ppa
需要输入用户密码,按照提示还需要按下 Enter 键
更新软件索引:
sudo apt update
四、软件和更新-附加驱动
可以看到开源的驱动(如果不添加PPA可能无法搜到),选择最新的驱动,然后点击应用更改,等待一会,这样驱动就更新好了。
注意:这个过程要联网!别中断!
五、重启、更改BIOS
重启,更改BIOS的Secure Boot为disabled,保存,不然主板拒绝加载第三方的显卡驱动,看看显卡驱动装成功没有,终端输入:
nvidia-smi
如果看到下图所示的信息,则说明显卡驱动安装成功,没有的话返回步骤4。
六、安装Anaconda,就可以开始创建环境了。
七、安装Pytorch(自带了cuda和cudnn,不需要自己下载,一条命令搞定)
但是官网下载特别慢,所以,我们需要换源。
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/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
换好源之后,把官网安装命令后面的 -c pytorch去掉,不然还是从pytorch官网下载就很慢了。
conda install pytorch torchvision torchaudio cudatoolkit=10.2
八、安装好Pytorch后测试一下能否调用GPU
python
import torch
torch.cuda.is_available()
结果返回True则安装成功!完美!