Hello Sanpreet Singh
I have gone through the error you are facing. I have faced such issues in the past. From my knowledge and experience the problem is with different versions of cuda, cuDNN and torch. If you are using RTX series GPU, cuda 9.0 have  some compatibility issues. So, it would be better you compile the code with cuda 10.0. I also recommend you please perform the task in a virtual environment. IF you don’t know about the virtual environment please follow these steps to create it.


Install pip first using this command:
sudo apt-get install python3-pip


Then install virtualenv using pip3:
sudo pip3 install virtualenv


Now create a virtual environment:
virtualenv venv (you can use any name instead of venv)


Active your virtual environment
source venv/bin/activate


Now, please visit https://pytorch.org/ 161 and install torch with cuda 10 .If you face any problem please have a look at the below link where i have added the screenshot for it. https://drive.google.com/file/d/1E0YmJGFBZ4b_P3a5kQotocLcKAvZCYJ_/view?usp=sharing 137
I am also adding the code if you don’t want to do it yourself.
pip3 install https://download.pytorch.org/whl/cu100/torch-1.0.1.post2-cp36-cp36m-linux_x86_64.whl

pip3 install torchvision

With regards
Ekta Smothra

  
      
      GitHub
  
  
    

EktaSmothra - Overview 27

EktaSmothra has 2 repositories available. Follow their code on GitHub.