CUDA 9.0 doesnt work with pycharm

I have CUDA 9.0 installed on my Ubuntu 18.04, and it success to compile on GCC.
however, when I opened a project with the pycharm 2016.3 (python 3.67), the checking line “torch.cuda.is_available()” return False.
do you have an idea how to fix it? maybe I have to define some configuration on the pycharm before i used the cuda?

thanks!

1 Like

probably your GPU driver is not installed correctly.

Follow the instructions in the linux install guide to verify your CUDA installation.

Also note that Ubuntu 18.04 is not an officially supported environment for CUDA 9.0. Unless you’ve done certain things like driver install carefully, it probably won’t work.