1.安装tensorflow-gpu
使用pip3安装tensorflow-gpu(版本1.14.0)
查看需要的配套程序(),网址:https://tensorflow.google.cn/install/source#bazel_build_options
2. Cuda安装
- cuda(10.0)下载
https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=runfilelocal
安装,按照官方网页即可安装。
由于没有root权限,安装到了/home/ljh/NN_install。
安装日志保存在了服务器/tmp/cuda-installer-19537中。
环境变量根据实际情况设置为了:
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/liujunhui/NN_install/lib64
export PATH=$PATH:/home/liujunhui/NN_install/bin
export CUDA_HOME=$CUDA_HOME:/home/liujunhui/NN_install/
source ~/.bashrc