30系显卡只支持cuda11.0及以上版本,但很多tensorflow项目用的仍然是1.1x版本,这些版本需要cuda10或者以下版本,这就导致在30系显卡上无法正常运1.1x版本的tensorflow,最近几天我也因为这个问题头疼不已,网上一番搜索之后发现了一个大神编译过后可用的版
下面是whl文件说明:
给RTX30系列显卡使用的tensorflow1.15的whl文件和cpp api,根据nvidia版的tf1.15改造编译,因为是魔改版,建议小心使用。
nvidia版默认开启了一些优化,需要小心的的关闭,这些优化会导致训练不正常以及性能问题,cpp api仅用于进行graph的推理。
我使用 python3.7 cuda11.2 cudnn8.1.0. 进行编译。
安装步骤:
1、首先准备一个带有RTX30系列显卡,然后安装好Python3.7环境建议使用anaconda3安装
2、安装好cuda11.2和cudnn
3、准备好whl文件,可以在gitee.com/FIRC/fircfiles/blob/master/tf1.15.4.md找到对应文件tensorflow-1.15.4+nv-cp37-cp37m-win_amd64.whl
4、切换到whl文件路径,执行pip install tensorflow-1.15.4+nv-cp37-cp37m-win_amd64.whl
注意事项:
必须是windows x64系统,只支持python3.7 64位环境,不支持其他python版本。tf1.x 官方已经不再支持了, 最新的tf1.15版本也停留在cuda10,而30系以上的新卡都不在支持cuda11之前的版本