InnovArul/nvidia_torch_issue.sh

Created April 29, 2025 20:41

Star (0) You must be signed in to star a gist
Fork (0) You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/InnovArul/620df960f22d03fa30b5a0e3b3f9c1a2.js"></script>
Save InnovArul/620df960f22d03fa30b5a0e3b3f9c1a2 to your computer and use it in GitHub Desktop.

Download ZIP

overcomes issue: RuntimeError: CUDA unknown error - this may be due to an incorrectly set up environment, e.g. changing env variable CUDA_VISIBLE_DEVICES after program start. Setting the available devices to be zero.

Raw

nvidia_torch_issue.sh

	sudo rmmod nvidia_uvm
	sudo modprobe nvidia_uvm

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment