Run Tensorflow with GPU in Pycharm

Viewed 1851

I want to run Tensorflow GPU in Pycharm on Windows 10, Cuda v11.1 and cuDNN v8.0.4.

I tried some guides like these

https://medium.com/@ashkan.abbasi/quick-guide-for-installing-python-tensorflow-and-pycharm-on-windows-ed99ddd9598

How to run Tensorflow GPU in Pycharm?

I installed https://developer.nvidia.com/cuda-downloads and https://developer.nvidia.com/rdp/cudnn-download.

I downloaded cudnn and unzipped it directly in the Cuda installation folder, is that correct? At C:\Program Files\NVIDIA GPU Computing Toolkit\

enter image description here


At the point 4- Install cuDNN Library on the medium blog I couldn't find the folder C:\Program Files\NVIDIA Corporation\NVSMI.

Do I have to install something extra to keep the folder, or was it renamed?


At the point 5- Install Tensorflow on the medium blog Tensorflow GPU is installed. I installed it with pip install tensorflow-gpu, but I don't have Anaconda Prompt.

Is there an option to run the gpu without installing Anaconda Prompt? How could I bypass the step?

1 Answers

You can install it in any Terminal without Anaconda prompt also using below line.

pip install tensorflow-gpu

You can follow this documentation about compatibility and Environment Variable setup.

You can also follow this post for installation using Visual Studio Express.

Related