-
Notifications
You must be signed in to change notification settings - Fork 2
Description
Line 69 in 3459515
| torch==1.8.1 |
When running python -m scripts.run_reals --type normal, I encounter the following warning: The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70.
If you want to use the NVIDIA GeForce RTX 3090 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/
As well as:
RuntimeError: CUDA error: no kernal image is available for the execution on the device.
Adding the suggestion of copying the installation notes from SUPPORT:
Install Pytorch with the version compatible with your OS and platform from https://pytorch.org/get-started/locally/
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
Running python -m scripts.run_reals --type normal correctly targets dedicated GPU for processing after performing the suggestion.