CODEWITHSUNDEEP
pythonpytorch
Matthew
Matthew

Reputation: 433

Runtime error 999 when trying to use cuda with pytorch

I installed Cuda 10.1 and the latest Nvidia Driver for my Geforce 2080 ti. I try to run a basic script to test if pytorch is working and I get the following error:

RuntimeError: cuda runtime error (999) : unknown error at ..\aten\src\THC\THCGeneral.cpp:50

Below is the code im trying to run:

import torch
torch.cuda.current_device()
torch.cuda.is_available()
torch.cuda.get_device_name(0)

Upvotes: 13

Views: 23376

Answers (4)

Dawid Laszuk
Dawid Laszuk

Reputation: 1978

In case of PyTorch it seems to be sufficient to restart only nvidia-uvm (Unified Virtual Memory) with

sudo modprobe --remove nvidia-uvm  # same as `rmmod`
sudo modprobe nvidia-uvm

If that doesn't work, go ahead and restart the whole module with additional modprobe (--remove) nvidia as mentioned @matwilso's answer.

Upvotes: 15

Qin Heyang
Qin Heyang

Reputation: 1674

Reinstalling NVIDIA driver solves this problem in my case. (You don't need to restart the computer)

Upvotes: 0

Julien Pilet
Julien Pilet

Reputation: 524

In my case, I solved the 999 error with: nvidia-modprobe -u

Upvotes: 3

matwilso
matwilso

Reputation: 3214

Restarting my computer fixed this for me.

But for a less invasive fix, you can also try this solution (from a tensorflow issue thread): 

sudo rmmod nvidia_uvm
sudo rmmod nvidia
sudo modprobe nvidia
sudo modprobe nvidia_uvm

Upvotes: 18

Related Questions