cuda10.1对应的tensorflow-gpu=2.2.0

电脑装了两个cuda版本,今天需要用到tensorflow-gpu=2.2.0,在环境变量里面已经把cuda10.1置顶了,但是怎么都调用不起来,在cmd里面输入nvcc -V显示的也是cuda10.1,但是就是用不了,只好把cuda装在conda环境里面了,输入下面这个就可以了

conda install cudatoolkit=10.1 cudnn=7.6.5

猜你喜欢

转载自blog.csdn.net/weixin_43850171/article/details/124436275
今日推荐