NVIDIA GeForce RTX 4090 with CUDA capability sm_89 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.
If you want to use the NVIDIA GeForce RTX 4090 GPU with PyTorch, please check the instructions at Start Locally | PyTorch
My OS is Ubuntu 18.04.01
Please help me solve this issue…
I can provide more information, if you tell me what Info I should provide
Based on the error message you have installed PyTorch binaries with CUDA 10.2, which is incompatible with your 4090.
Install the latest or nightly release with CUDA 11.7 or 11.8 (I would recommend using the nightly release with CUDA 11.8 for the best performance) using the install commands from here.
How did you figure out it was cuda 10.2 earlier?
I did the following and solved my issue:
conda remove pytorch
conda remove cudatoolkit
Then the command from pytorch instructions from your link
Thank you
The supported architectures were reported as:
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.
which we used in all CUDA 10 builds. Unless you were using a custom build and set the architectures manually, your wheels were most likely using CUDA 10.2. 