'conda install tensorflow-gpu' is not installing any CUDNN libraries

user_7789223 0
Viewed 3513

When I run conda install tensorflow-gpu in a new venv, conda gives the list of packages that it will install, but there are no CUDNN or cudatoolkit packages which I would expect, and that other people have. I am expecting an output like in this question (Is it still necessary to install CUDA before using the conda tensorflow-gpu package? ) but instead I get this:

The packages it is installing for me, I'm expecting CUDNN

Edit: I was able to get it to work by running this command instead

conda create -n tf-gpu python=3.7 anaconda followed by conda install tensorflow-gpu

After doing this it installed all the cudnn libraries on its own and I was able to run tensorflow with GPU. However, for some reason it is installing tensorflow 1.14, and I need 2.x so does anyone know why this is?

anaconda tensorflow gpu
1 Answers

conda install tensorflow-gpu install only gpu version of Tensorflow and not supporting packages such as CUDA and cuDNN. Follow steps to install

conda install cudatoolkit=10.0.130
conda install cudnn=7.6.0=cuda10.0_0
conda install tensorflow-gpu
Admin1
Related