+关注
已关注

分类  

暂无分类

标签  

暂无标签

日期归档  

2019-08(47)

2019-09(90)

2019-10(13)

2019-11(9)

2019-12(19)

Ubuntu 18.04 CUDA10.1 Anaconda 安装tensorflow-gpu 1.13.1

发布于2020-08-11 16:47     阅读(514)     评论(0)     点赞(19)     收藏(4)


由于想使用https://github.com/matterport/Mask_RCNN

这个代码使用的是比较老版本的tensorflow 对应的cuda版本是10.0或9.0或9.2,我系统安装的cuda是10.1版本

cudatoolkit是库不是驱动,库意味着不是必须安装到系统目录,可以安装到用户目录,具体链接哪些库,怎么链接由Makefile决定,可执行文件执行的时候,在LD_LIBRARY_PATH路径或其他路径里能找到需要的库文件即可。

查看LD_LIBRARY_PATH方法

echo $LD_LIBRARY_PATH

查看nvidia驱动是否安装好

  1. nvidia-smi
  2. nvcc -v

在系统安装好驱动后,anaconda安装cudatoolkit是很方便的,需要配置好源,我使用的源如下(~/.condarc):

  1. ssl_verify: true
  2. channels:
  3. - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
  4. - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  5. - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  6. - https://mirrors6.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
  7. - https://mirrors6.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  8. - https://mirrors6.tuna.tsinghua.edu.cn/anaconda/cloud//pytorch/
  9. show_channel_urls: true

安装cudatoolkit命令

conda install cudatoolkit=10.0

安装tensorflow-gpu命令

  1. #1
  2. conda install tensorflow-gpu=1.13.1
  3. #2
  4. pip install tensorflow-gpu=1.13.1

/home/kitty/anaconda3/envs/py36/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py:526: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'.
  _np_qint8 = np.dtype([("qint8", np.int8, 1)])

出现上面警告是由于,tensorflow-gpu 1.13.1要求numpy版本不能太高,可以使用下面命令解决。

  1. pip uninstall numpy
  2. pip install numpy==1.16.4

测试GPU对安装的tensorflow是否可用

  1. import tensorflow as tf
  2. print(tf.test.is_gpu_available())

 

如果返回True,接下来tensorflow-gpu就可以正常使用了。

安装好后python库列表

  1. (py36) kitty@heal-PowerEdge-T630:~$ conda list
  2. # packages in environment at /home/kitty/anaconda3/envs/py36:
  3. #
  4. # Name Version Build Channel
  5. _libgcc_mutex 0.1 main https://mirrors6.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  6. blas 1.1 openblas https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
  7. ca-certificates 2020.6.20 hecda079_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
  8. certifi 2020.6.20 py36h9f0ad1d_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
  9. cudatoolkit 10.0.130 0 https://mirrors6.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  10. future 0.18.2 pypi_0 pypi
  11. keras-preprocessing 1.1.2 pypi_0 pypi
  12. ld_impl_linux-64 2.33.1 h53a641e_7 https://mirrors6.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  13. libedit 3.1.20191231 h14c3975_1 https://mirrors6.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  14. libffi 3.3 he6710b0_2 https://mirrors6.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  15. libgcc-ng 9.1.0 hdf63c60_0 https://mirrors6.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  16. libgfortran-ng 7.5.0 hdf63c60_14 https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
  17. libstdcxx-ng 9.1.0 hdf63c60_0 https://mirrors6.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  18. mock 4.0.2 pypi_0 pypi
  19. ncurses 6.2 he6710b0_1 https://mirrors6.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  20. numpy 1.16.4 pypi_0 pypi
  21. openblas 0.3.3 h9ac9557_1001 https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
  22. openssl 1.1.1g h516909a_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
  23. pip 20.2.1 py36_0 https://mirrors6.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  24. python 3.6.10 h7579374_2 https://mirrors6.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  25. python_abi 3.6 1_cp36m https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
  26. readline 8.0 h7b6447c_0 https://mirrors6.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  27. setuptools 49.2.1 pypi_0 pypi
  28. sqlite 3.32.3 h62c20be_0 https://mirrors6.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  29. tensorflow-gpu 1.13.1 pypi_0 pypi
  30. tk 8.6.10 hbc83047_0 https://mirrors6.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  31. wheel 0.34.2 pypi_0 pypi
  32. xz 5.2.5 h7b6447c_0 https://mirrors6.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  33. zlib 1.2.11 h7b6447c_3 https://mirrors6.tuna.tsinghua.edu.cn/anaconda/pkgs/main

 

参考文献:

1.FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'.解决办法

2.官网经过测试的构建配置

3.非ROOT用户安装CUDA和CUDNN

4.如何查看tensorflow-gpu是否可用



所属网站分类: 技术文章 > 博客

作者:dfh8374

链接: https://www.pythonheidong.com/blog/article/479213/

来源: python黑洞网

任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任

19 0
收藏该文
已收藏

评论内容:(最多支持255个字符)