tensorflow2.0升级,更换Ubuntu16.04驱动以及CUDA、CUDNN
之前:
显卡 GTX960M
tensorflow-gpu:2.0
cuda:10.0
cudnn:7.4
因为代码要求需要使用tensorflow-gpu:2.0以上版本,所以需要更换环境
可以看到version是460.84,之前的是418,不满足更换tensorflow2.1cuda10.1的要求:
更换驱动以及内核
在2021年1月26日,英伟达非常良心地更新驱动,让GTX960M也支持cuda11了,所以我们可以更换驱动,便可以更换CUDA了(CUDA版本对应下图红线,需要根据你的显卡内核来确
原创
2021-06-20 09:48:10 ·
455 阅读 ·
0 评论