CUDA available in notebook but not in VS code terminal - same conda environment

Question

I have torch installed in a conda environment. When I run python in command line and try to check CUDA availability I get False as show below.

>>> import torch
>>> torch.cuda.is_available()
False

However, when I use jupyter notebook, it shows CUDA is available. I'm using the same environment. The GPU is on the same local Windows machine

import torch
torch.cuda.is_available()
> True
!which python
> /c/Users/b2bt/anaconda3/envs/stable-diffusion/python

I tried creating a new conda environment to run the stable diffusion web UI. I've installed torch previously in a different environment and have used it successfully with GPU. I was expecting the terminal to show CUDA available but strangely it doesn't.

MingJie-MSFT · Accepted Answer · 2023-05-15 01:33:58Z

2

This problem may be caused by using pip install pytorch to install package.

Solution:

Open the official website, and search for the correct version for your environment.

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117

answered May 15, 2023 at 1:33

MingJie-MSFT

10.1k1 gold badge11 silver badges30 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

kamal tanwar Over a year ago

I never got to try this solution but thanks for answering. Previously I was using the .bat file to install everything which was provided in the repository. But then I tried installing everything manually and it worked.

Sughosh Kulkarni Over a year ago

Thanks! I used pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117 and it worked for me.

Asha Choudhary · Accepted Answer · 2023-08-04 09:56:35Z

2

I had the same issue as you have. I realized pytorch was not installed.

So, installed in using pip3 command stated above:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117

answered Aug 4, 2023 at 9:56

Asha Choudhary

1631 silver badge6 bronze badges

Collectives™ on Stack Overflow

CUDA available in notebook but not in VS code terminal - same conda environment

2 Answers 2

2 Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

2 Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related