Ubuntu Server 18.04 搭建深度学习平台

2019独角兽企业重金招聘Python工程师标准>>>

1、设置python版本

sudo update-alternatives --install /usr/bin/python python /usr/bin/python3 150

2、安装Anaconda虚拟环境管理

wget https://repo.anaconda.com/archive/Anaconda3-5.3.0-Linux-x86_64.sh

# bash Anaconda3-5.2.0-Linux-x86_64.sh

记得把环境变量添加在系统中，最后是否安装VScode建议不安装

3、安装CUDA和cuDNN，在版本选择上因为GPU原因选择CUDA10.0和相应的cuDNN

选择对应系统的版本，在线安装：https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=debnetwork

https://developer.nvidia.com/rdp/cudnn-download需要注册登录账号，下载后解压

sudo cp cuda/include/cudnn.h    /usr/local/cuda/include      
sudo cp cuda/lib64/libcudnn*    /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h   /usr/local/cuda/lib64/libcudnn*

4、下载并安装GPU驱动

https://www.nvidia.cn/Download/index.aspx?lang=cn

lsmod | grep nvidia，如果有NVIDIA驱动，请先卸载

lsmod | grep nouveau，如果有其他驱动，将其他显卡驱动加入黑名单，操作如下：

cd /etc/modprobe.d/

创建blacklist-nouveau.conf并添加内容

sudo touch blacklist-nouveau.conf

sudo echo "blacklist nouveau">>blacklist-nouveau.conf

sudo echo "options nouveau modeset=0">>blacklist-nouveau.conf

查看blacklist-nouveau.conf内容是否添加成功

sudo cat blacklist-nouveau.conf

sudo update-initramfs -u

现在请重启机器，使用init 3进入命令行界面

sudo chmod +x NVIDIAXXXXXXXXX.run

sudo ./NVIDIAXXXXXXXX.run -no-opengl-files 完成安装

nvidia-smi检查显卡是否在工作

5、创建conda虚拟环境，安装Tensorflow或者Pytorch

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free

conda config --set show_channel_urls yes

conda create --name XXXX python=3.X

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

到这里，你的深度学习平台已经搭建完成，你可以把它分享给其他人使用了。

不足之处，还望指正，谢谢！！

 

 

 

转载于:https://my.oschina.net/aisi746467512/blog/3016906