Ubuntu19.10安装Nvidia驱动,cuda+cudnn,anaconda和pytorch-gpu
1. nvidia驱动安装
- ubuntu19.10提供了nvidia官方驱动的安装方式,打开软件《附加驱动》,安装推荐的nvidia驱动;
- 选择“应用更改”,完成后重启;(这个过程可能出现开机进入grub,找不到引导,只要重启进入bios设置win10为第一启动项即可(默认进入win10系统),随后“esc”选择ubuntu引导启动)
2. cuda和cudnn的安装
- 首先确定安装的cuda版本,去这里查看cuda版本兼容的nvidia驱动,我的nvidia驱动是435,所以选择cuda10.1
- -或者直接终端输入nvidia-smi查看cuda信息,右上角有显示的cuda版本
- 接下来终端输入sudo apt install cuda 直接安装官方的cuda库,这里会检查电脑的nvidia版本,所以不用担心安装的什么cuda版本,也不用再去显示的安装cudnn
3. anaconda的安装
bash Anaconda3-2020.02-Linux-x86_64 (1).sh
- 也可以安装在/root下(命令:sudo sh Anaconda3-2020.02-Linux-x86_64 (1).sh ),但是需要sudo -i 在管理员模式下运行(很麻烦);
- 安装过程中一路yes,环境变量也不需要手动添加;
4. pytorch-gpu版本安装
conda install --use-local pytorch-1.4.0-py3.7_cuda10.1.243_cudnn7.6.3_0.tar.bz2
- 然后进入anaconda环境,输入python,在python环境下检测pytorch安装是否成功
import torch
torch.cuda.is_available()