ubuntu 18.04+pytorch+cuda10.0.130+cudnn10.0

最新推荐文章于 2023-03-16 14:19:13 发布

pycharD

最新推荐文章于 2023-03-16 14:19:13 发布

阅读量1.3k

点赞数

分类专栏： # Ubuntu 文章标签： ubuntu

原文链接：https://www.cnblogs.com/zhengbiqing/p/10346212.html

版权

Ubuntu 专栏收录该内容

7 篇文章 1 订阅

订阅专栏

ubuntu 18.04+pytorch+cuda10.0.130+cudnn10.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

sudo sh cuda_10.0.130_410.48_linux.run	　　#下载cuda_10.0.130_410.48_linux.run
nvidia-smi		#查看显卡信息

添加环境变量：
vi ~/.bashrc #打开文件
export CUDA_HOME=/usr/local/cuda
export PATH= $P A T H :$ CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64 $KaTeX parse error: Expected '}', got 'EOF' at end of input: \dotsLIBRARY_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 vim ~/.bashrc
　　添加 export LD_LIBRARY_PATH= $H O M E / c u d a / l i b 64 :$ 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

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

REFERENCE: https://www.cnblogs.com/zhengbiqing/p/10346212.html