CUDA not being recognized in jupyter notebook but recognized when running .py files I have a problem with CUDA and PyTorch.