版权声明:本文为博主原创文章,欢迎转载,并请注明出处。联系方式:460356155@qq.com

ubuntu 16.04用了1年多了,18.04版已经发布也半年了,与时俱进,重装Linux系统,这里主要记录下gpu加速pytorch 1.0.0版本的安装。

一、工具安装

  sudo apt-get install gcc

  sudo apt-get install g++

  sudo apt-get install make

二、禁用nouveau

  sudo gvim /etc/modprobe.d/blacklist.conf

  添加 blacklist nouveau

  sudo update-initramfs -u

  sudo reboot

可以通过  lsmod | grep nouveau   检查是否已禁用。

三、安装cuda

  下载cuda_10.0.130_410.48_linux.run

  sudo sh cuda_10.0.130_410.48_linux.run

  nvidia-smi查看显卡信息

  编辑~/.bashrc,添加行

  export CUDA_HOME=/usr/local/cuda

  export PATH=$PATH:$CUDA_HOME/bin

  export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

  即时生效环境变量:source ~/.bashrc

  检查是否安装成功的几个方法:

  1、nvcc -V

  2、cat /proc/driver/nvidia/version

  3、cd /usr/local/cuda/samples/1_Utilities/deviceQuery

    sudo make

    ./deviceQuery

四、安装cudnn 10.0

  下载cudnn-10.0-linux-x64-v7.4.2.24.tgz

   tar xzvf cudnn-10.0-linux-x64-v7.4.2.24.tgz

  mv cuda/ ~/cuda

  sudo gvim ~/.bashrc

  添加 export LD_LIBRARY_PATH=$HOME/cuda/lib64:$LD_LIBRARY_PATH

  即时生效环境变量:source ~/.bashrc

五、安装pytorch

  sudo apt install python3-pip

  sudo pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple numpy

  sudo pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple matplotlib

  sudo pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple opencv-python

  sudo pip3 install https://download.pytorch.org/whl/cu100/torch-1.0.0-cp36-cp36m-linux_x86_64.whl

  sudo pip3 install torchvision

六、检查是否安装成功

  import torch

  print(torch.__version__)

  print(torch.cuda.is_available())

以上是pytorch 1.0、cuda、cudnn的安装过程。

jupyter notebook是python开发的很好的工具,安装如下:

  sudo apt-get install jupyter-notebook

  sudo jupyter-notebook --generate-config –allow-root

  自动生成  ~/.jupyter/jupyter-notebook-config.py


  创建目录 /home/<当前用户>/work'

  用文本编辑器编辑刚才生成的配置文件jupyter-notebook-config.py,修改文件中如下参数

  c.NotebookApp.notebook_dir = '/home/<当前用户>/work'

  c.NotebookApp.token='登陆密码'  如果设置为空'',则运行则不需要密码

  运行noteboook:

  sudo jupyter-notebook --allow-root

  浏览器访问:

  http://localhost:8888

  停止notebook:

  sudo jupyter-notebook stop


-->

ubuntu 18.04安装pytorch、cuda、cudnn等的更多相关文章

  1. 真实机下 ubuntu 18.04 安装GPU +CUDA+cuDNN 以及其版本选择(亲测非常实用)【转】

    本文转载自:https://blog.csdn.net/u010801439/article/details/80483036 ubuntu 18.04 安装GPU +CUDA+cuDNN : 目前, ...

  2. [笔记] Ubuntu 18.04安装Docker CE及nvidia-docker2流程

    Docker的好处之一,就是在Container里面可以随意瞎搞,不用担心弄崩Host的环境. 而nvidia-docker2的好处是NVidia帮你配好了Host和Container之间的CUDA相 ...

  3. [笔记] Ubuntu 18.04安装Docker CE及NVIDIA Container Toolkit流程

    之前写的[笔记] Ubuntu 18.04安装Docker CE及nvidia-docker2流程已经out了,以这篇为准. Docker的好处之一,就是在Container里面可以随意瞎搞,不用担心 ...

  4. Ubuntu 18.04 安装MySQL

    最近在写东西的时候,需要用到MySQL,在网上查了一下,都说Ubuntu18.04不能安装MySQL5.7.22, 总觉的不可能,所以自己就研究了一下,然后分享给大家 工具/原料   VMware W ...

  5. Ubuntu 18.04 安装博通(Broadcom)无线网卡驱动

    目录 Ubuntu 18.04 安装博通(Broadcom)无线网卡驱动 Package gcc is not configured yet. 解决办法 history history | grep ...

  6. 在Ubuntu 18.04 安装 MySQL 8.0

    在Ubuntu 18.04 安装 MySQL 8.0 ① 登入 mysql 官网,在官网中下载 deb 包,点击该链接,即可下载. https://dev.mysql.com/downloads/re ...

  7. Ubuntu 18.04 安装部署Net Core、Nginx全过程

    Ubuntu 18.04 安装部署Net Core.Nginx全过程 环境配置 Ubuntu 18.04 ,Nginx,.Net Core 2.1, Let's Encrypt 更新系统 sudo a ...

  8. Ubuntu 18.04 安装和常用软件安装

    Ubuntu 18.04 安装 下载 Ubuntu 制作 U 盘启动盘 设置电脑为 U 盘启动 插入 U 盘,重启电脑 按照提示安装 Ubuntu 更新 NVIDIA 显卡和 Broadcom 无线网 ...

  9. Ubuntu 18.04安装MongoDB 4.0(社区版)

    Ubuntu 18.04(虚拟机VirtualBox上),MongoDB 4.0, 听室友说,23点有世界杯决赛呢!可是,孤要写博文的啊!以记录这忙乱的下午和晚间成功安装了一个软件到Linux上.—— ...

随机推荐

  1. 触发器 评论折叠显示(jquery)

    触发器   评论折叠显示(jquery)   子页面 : @if(Model?.count>0) {      var i=1;      var n=model.count-2      va ...

  2. C#多线程编程的同步也线程安全

    前一篇文章记录了简单的多线程编程的几种方式,但是在实际的项目中,也需要等待多线程执行完成之后再执行的方法,这个就叫做多线程的同步,或者,由于多个线程对同一对象的同时操作造成数据错乱,需要线程安全.这篇 ...

  3. 学JAVA第三天,JAVA第二章《JAVA数据类型》

    ---恢复内容开始--- <JAVA数据类型> 我们一般都用int类型,因为int类行一般的日常生活的数据都能满足了. 当然,想李嘉诚,马云这种有钱人,int类行就不能满足帮他记钱的了,像 ...

  4. Hibernate框架笔记03表操作多对多配置

    目录 1. 数据库表与表之间的关系 1.1 一对多关系 1.2 多对多关系 1.3 一对一关系[了解] 2. Hibernate的一对多关联映射 2.1 创建一个项目,引入相关jar包 2.2. 创建 ...

  5. 微信小程序异步请求问题

    微信小程序为了提高用户体验,提供的api大部分都是异步操作,除了数据缓存操作里面有一些同步操作.是提高了用户体验,但是在开发的时候, 就有点坑了,例如我要写一个公共方法,发起网络请求,去后台去一些数据 ...

  6. vue.js引入

    开始学习vue.js,引入vue.vue.js一定要在head里面引入,实际开发中我们可能在body中引入,但是可能存在抖屏现象. 为了避免出现抖屏现象,我们引入vue.js或者jquery.js 最 ...

  7. Dynamics 365 Customer Engagement安装FAQ

    微软动态CRM专家罗勇 ,回复310或者20190308可方便获取本文,同时可以在第一间得到我发布的最新博文信息,follow me!我的网站是 www.luoyong.me . 本文参考了包括但不限 ...

  8. How to Apply Patches to a WLS 8.1 Environment

    APPLIES TO: Oracle Weblogic Server - Version 8.1 to 8.1Information in this document applies to any p ...

  9. weblogic Patch

    How to Apply WLS Patch on Weblogic Integrated with OEM Getting "Main Thread" Java.lang.Out ...

  10. 基础环境系列:Apache2.4.37

    一.安装 进入官网http://www.apache.org/,滑至最下方,排名第一的HTTP Server就是我们需要的. 当前时间的最新版本是2.4.37.呃……并没有msi版本,我们选择最后一个 ...