如何在Ubuntu18.10下搭建深度学习环境

2024-11-03 16:59:03

1、安装cuda之前的准备降级gcc版本为 4.8,g++版本为 4.8sudo apt install gcc-4.8 g++-4.8sudo update-alternativ髫潋啜缅es --install /usr/bin/gcc gcc /usr/bin/gcc-4.8 100sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.8 80

2、.安装cuda 9.0(1)下载16.04版本对应的cuda的run文件,安装选项如图1所示 (2)下载完成后,在终端打开该下载总目录,输入sudo sh sudo shcuda_9.0.176_384.81_linux.run。 (3)根据提示输入accept,y,Enter等

3、设置cuda环境变量sudo gedit ~/.bashrc文本最后添加内容Export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:/usr/local/cuda/extras/CPUTI/lib64export CUDA_HOME=/usr/local/cuda-9.0/binexport PATH=$PATH:$LD_LIBRARY_PATH:$CUDA_HOME

4、检测cuda:cd /usr/local/cuda-9.0/samples/1_Utilities/deviceQuerysudo make./deviceQuery看到最后结果是:Result =PASS如图即可

如何在Ubuntu18.10下搭建深度学习环境

6、安装tensorflow-gpu$pip3install--upgrade tensorflow-gpu==1.10.0(python3.6)$Pip install--upgrade tensorflow-gpu==1.10.0( python2.7 )测试安装是否成功,终端中输入Python或python3再输入import tensorflow as tf无误则安装成功

猜你喜欢