Ubuntu19.10安装Nvidia驱动,cuda+cudnn,anaconda和pytorch-gpu

Ubuntu19.10安装Nvidia驱动,cuda+cudnn,anaconda和pytorch-gpu

1. nvidia驱动安装

  • ubuntu19.10提供了nvidia官方驱动的安装方式,打开软件《附加驱动》,安装推荐的nvidia驱动;
    ubuntu安装- 选择“应用更改”,完成后重启;(这个过程可能出现开机进入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()
发布了2 篇原创文章 · 获赞 0 · 访问量 58

猜你喜欢

转载自blog.csdn.net/weixin_42025760/article/details/105141267