ubuntu16.04+cuda10.1+cudnn7.6.4+tensorflow-gpu环境配置

本文作为环境配置的记录。
在安装完显卡驱动后,得知需要的cuda为10.1,可以在这个网站找到合适版本的cuda:
https://developer.nvidia.com/cuda-toolkit-archive
安装过程参考
https://www.cnblogs.com/dinghongkai/p/11268976.html
但是事实上安装的过程中没有关闭图形化界面,直接在终端运行了.run也是可以的。

然后找到对应版本的cudnn:
https://developer.nvidia.com/rdp/cudnn-archive
下载之后(参考:https://blog.csdn.net/wanzhen4330/article/details/81699769)

	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

	sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

cudnn安装完成后,安装tensorflow-gpu
直接使用pip3太慢,使用清华源出现了报错,使用豆瓣的源再次安装。
报错1:

ERROR: twisted 18.7.0 requires PyHamcrest>=1.9.0, which is not installed.

先安装PyHamcrest,再安装,解决。
报错2:

ERROR: Cannot uninstall 'wrapt'. It is a distutils installed project and thus we cannot accurately determine which files belong to it which would lead to only a partial uninstall.

安装wrapt即可

	sudo pip install tensorflow --ignore-installed wrapt 

报错3:

ERROR: Exception: Traceback (most recent call last)

网速问题,多次换源无果后,下载了whl文件安装。
下载地址:https://pypi.org/project/tensorflow-gpu/1.15.2/#files

报错4:
(具体问题暂时忘记,留个坑)
原因是使用了最新的pip20.0.2,卸载安装了19版本,解决。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值