這是同事遇到的一個問題,就是在服務器上的虚拟环境envA中 明明安裝了 tensorflow-gpu 但是卻始終無法檢測到cuda 一開始懷疑是因為 cuda 或者 CUDnn這一部分有問題 後面檢查之後發現完好 驅動也正常。
於是使用命令
pip -V
python -V
发现虽然我们已经通过 conda activate envA 进入了环境里面了 但是pip 指向的是 原生的python(不是这个虚拟环境的python)于是我们最后使用:
python -m install tensorflow-gpu==2.6.0
才终于安装成功了
当然我们也可以直接在jupyter notebook 里面使用
!python -m install tensorflow-gpu
对这个环境进行安装