ubuntu 18.04安装pytorch、cuda、cudnn等
版权声明:本文为博主原创文章,欢迎转载,并请注明出处。联系方式: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等的更多相关文章
- 真实机下 ubuntu 18.04 安装GPU +CUDA+cuDNN 以及其版本选择(亲测非常实用)【转】
本文转载自:https://blog.csdn.net/u010801439/article/details/80483036 ubuntu 18.04 安装GPU +CUDA+cuDNN : 目前, ...
- [笔记] Ubuntu 18.04安装Docker CE及nvidia-docker2流程
Docker的好处之一,就是在Container里面可以随意瞎搞,不用担心弄崩Host的环境. 而nvidia-docker2的好处是NVidia帮你配好了Host和Container之间的CUDA相 ...
- [笔记] Ubuntu 18.04安装Docker CE及NVIDIA Container Toolkit流程
之前写的[笔记] Ubuntu 18.04安装Docker CE及nvidia-docker2流程已经out了,以这篇为准. Docker的好处之一,就是在Container里面可以随意瞎搞,不用担心 ...
- Ubuntu 18.04 安装MySQL
最近在写东西的时候,需要用到MySQL,在网上查了一下,都说Ubuntu18.04不能安装MySQL5.7.22, 总觉的不可能,所以自己就研究了一下,然后分享给大家 工具/原料 VMware W ...
- Ubuntu 18.04 安装博通(Broadcom)无线网卡驱动
目录 Ubuntu 18.04 安装博通(Broadcom)无线网卡驱动 Package gcc is not configured yet. 解决办法 history history | grep ...
- 在Ubuntu 18.04 安装 MySQL 8.0
在Ubuntu 18.04 安装 MySQL 8.0 ① 登入 mysql 官网,在官网中下载 deb 包,点击该链接,即可下载. https://dev.mysql.com/downloads/re ...
- 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 ...
- Ubuntu 18.04 安装和常用软件安装
Ubuntu 18.04 安装 下载 Ubuntu 制作 U 盘启动盘 设置电脑为 U 盘启动 插入 U 盘,重启电脑 按照提示安装 Ubuntu 更新 NVIDIA 显卡和 Broadcom 无线网 ...
- Ubuntu 18.04安装MongoDB 4.0(社区版)
Ubuntu 18.04(虚拟机VirtualBox上),MongoDB 4.0, 听室友说,23点有世界杯决赛呢!可是,孤要写博文的啊!以记录这忙乱的下午和晚间成功安装了一个软件到Linux上.—— ...
随机推荐
- 触发器 评论折叠显示(jquery)
触发器 评论折叠显示(jquery) 子页面 : @if(Model?.count>0) { var i=1; var n=model.count-2 va ...
- C#多线程编程的同步也线程安全
前一篇文章记录了简单的多线程编程的几种方式,但是在实际的项目中,也需要等待多线程执行完成之后再执行的方法,这个就叫做多线程的同步,或者,由于多个线程对同一对象的同时操作造成数据错乱,需要线程安全.这篇 ...
- 学JAVA第三天,JAVA第二章《JAVA数据类型》
---恢复内容开始--- <JAVA数据类型> 我们一般都用int类型,因为int类行一般的日常生活的数据都能满足了. 当然,想李嘉诚,马云这种有钱人,int类行就不能满足帮他记钱的了,像 ...
- Hibernate框架笔记03表操作多对多配置
目录 1. 数据库表与表之间的关系 1.1 一对多关系 1.2 多对多关系 1.3 一对一关系[了解] 2. Hibernate的一对多关联映射 2.1 创建一个项目,引入相关jar包 2.2. 创建 ...
- 微信小程序异步请求问题
微信小程序为了提高用户体验,提供的api大部分都是异步操作,除了数据缓存操作里面有一些同步操作.是提高了用户体验,但是在开发的时候, 就有点坑了,例如我要写一个公共方法,发起网络请求,去后台去一些数据 ...
- vue.js引入
开始学习vue.js,引入vue.vue.js一定要在head里面引入,实际开发中我们可能在body中引入,但是可能存在抖屏现象. 为了避免出现抖屏现象,我们引入vue.js或者jquery.js 最 ...
- Dynamics 365 Customer Engagement安装FAQ
微软动态CRM专家罗勇 ,回复310或者20190308可方便获取本文,同时可以在第一间得到我发布的最新博文信息,follow me!我的网站是 www.luoyong.me . 本文参考了包括但不限 ...
- 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 ...
- weblogic Patch
How to Apply WLS Patch on Weblogic Integrated with OEM Getting "Main Thread" Java.lang.Out ...
- 基础环境系列:Apache2.4.37
一.安装 进入官网http://www.apache.org/,滑至最下方,排名第一的HTTP Server就是我们需要的. 当前时间的最新版本是2.4.37.呃……并没有msi版本,我们选择最后一个 ...