Need to run torch 1.9.0 >= 1.6.0 on RTX4090 because these are the requirements for Unity mlagents
What I need to figure out is what version of Cuda to install to satisfy these requirements on Ubuntu 20.04
I’ve tried Cuda Version 12.7 and 11.6 with torch 1.6.0 but I still see the same error:
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_70 sm_75
CUDA 12.7 is not a thing and your locally installed CUDA toolkit won’t be used (unless you build PyTorch from source or a custom CUDA extension). Based on the error message you have installed a PyTorch binary with CUDA 10.2 runtime dependencies and you can verify it via print(torch.version.cuda). If so, install a PyTorch binary with CUDA 11+ runtime dependencies.