终于自己的电脑上NVIDIA显卡了。
Ubuntu下配置TensorFlow真是遇到了大坑
GPU:NVIDIA 1660ti
系统:Ubuntu 16.04
显卡驱动:418.43
cuda:9.0
cudnn:7.3.1
tensorflow:1.12.0
真是大坑啊,tensorflow从1.13开始要依赖cuda10.0了,所以千万不要用最新的tensorflow,建议1.12。
tensorflow版本1.12,所以cuda得用9.0,cudnn一定要7.3.1,我试了7.5和7.4,都会报:Failed to get convolution algorithm. This is probably because cuDNN failed to initialize 这个错误,大坑
1660ti是新卡,驱动版本最少是418。