Yuw*_*Yan 3 cuda tensorflow cudnn
当前,我需要在存在较旧版本CUDA和cuDNN的计算机上安装gpu版本的tensorflow,我的问题是,如何在不重新安装CUDA和cuDNN的情况下完成它,因为其他软件包(请参阅pytorch)需要它。
根据这篇文章,使用以下命令检查CUDA版本。
$ nvcc --version
Run Code Online (Sandbox Code Playgroud)
根据这篇文章,使用以下命令检查cuDNN版本。
$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
Run Code Online (Sandbox Code Playgroud)
根据这篇文章选择tensorflow-gpu版本