Description
Can’t use PyTorch with GPU Nvidia A100.
Get this error message
NVIDIA A100-PCIE-40GB with CUDA capability sm_80 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70.
If you want to use the NVIDIA A100-PCIE-40GB GPU with PyTorch, please check the instructions at Start Locally | PyTorch
Environment
TensorRT Version: NA
GPU Type: NVIDIA A100-PCIE-40GB
Nvidia Driver Version:510.47.03
CUDA Version: 11.6
CUDNN Version: NA
Operating System + Version: rhel + 8.5
Python Version (if applicable): 3.9.12
TensorFlow Version (if applicable):
PyTorch Version (if applicable): 1.12.0
Baremetal or Container (if container which image + tag): baremeta