您的位置:首页 > 其它

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

2020-04-02 19:07 609 查看

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的安装

  • 去清华镜像下载安装包,链接为:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
  • 选好对应的版本,我的是Anaconda3-2020.02-Linux-x86_64 (1).sh ,然后终端执行如下命令,默认安装在/home/用户名 下;
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版本安装

  • 同样清华镜像https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/下载需要的pytorch安装包,本人是pytorch-1.4.0-py3.7_cuda10.1.243_cudnn7.6.3_0.tar.bz2
  • cd到安装包所在目录,进行本地安装:
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()
  • 点赞
  • 收藏
  • 分享
  • 文章举报
TheoRenli 发布了1 篇原创文章 · 获赞 0 · 访问量 22 私信 关注
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐