环境:(亲测可用)
window10
RTX3090
python3.7.9
cuda11.0, cuda11.1(推荐,将bin文件中cusolver64_11.dll,修改为cusolver64_10.dll即可使用,更加方便快捷。)
cudnn-11.0-windows-x64-v8.0.4.30
tensorflow-gpu=2.4.0rc2
或者直接安装 pip install tf-nightly-gpu -i http://pypi.douban.com/simple --trusted-host pypi.douban.com (cuda11.1+cudnn8.1.0)
请先获得下面的ptxas.exe,替换C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin\ptxas.exe.
问题一:如果报缺cufft64_10.dll或者cudnn64_8.dll或者curand64_10.dll或者cusolver64_10.dll,或者cudart64_101.dll等等,下载相应dll文件放到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin中
问题二:SubProcess ended with return code: 4294967295
复制C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin\ptxas.exe 到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin\ptxas.exe.
通过CSDN资源搜索获得cuda11.1的ptxas.exe : https://download.csdn.net/download/qq_38784454/13773286,或者百度网盘链接:https://pan.baidu.com/s/1-WCGpN9QvDErEC16VXq5UA
提取码:1234