小白,第一次搭这个环境,花了很长时间,先是搭的cpu版本的,后来又尝试了gpu版本的,中间也有各种错误,这篇文章是tensorflow-gpu的,其实我觉得差不多,记录一下大致过程~~
代码
用的是这个代码: https://github.com/dBeker/Faster-RCNN-TensorFlow-Python3
环境配置
cuda
首先要注意显卡驱动和cuda的版本要对应
我装的是10.0的版本
附上链接: https://developer.nvidia.com/cuda-10.0-download-archive
下完之后运行exe文件进行安装就可以了,具体每一步是怎么样的也可以搜一下其他教程
cudnn
地址: https://developer.nvidia.com/rdp/cudnn-archive
要登陆,然后自己选择要安装的版本,我装的是7.4.2 for cuda10.0的那一个
下载后解压得到几个文件夹,将文件夹里面的内容对应复制到cuda对应的文件夹下