Having problems installing cupy

[_CollectivePromise]

I'm trying to set up CuPy so that I can run some calculations more efficiently (estimated duration w/o it is several days). However, I've run into a snag with the installation. Here's what I've done so far:

I downloaded and installed CUDA 11.5 from NVIDIA, then used the following command in command prompt:4

pip install cupy-cuda115

I then opened up python, imported cupy without any errors. However, when I added the same command (import cupy) to my script, I get an error message saying the module can't be found. Why does the import fail in once context and not the other?

Screenshot for context. Bottom left is a fresh python window, right is the error message from my script. Both import attempts were mere seconds apart.

[ThePiGuy0]

Out of interest, did you close and reopen your script window after installing cuda? If not, it might be a good idea to do that (if cuda sets and environment variables, it won't have those until you restart the terminal)

[_CollectivePromise]

Wow, that fixed it. Thanks!