Hi, I installed torch 1.8.1 through conda and have RTX 3080 GPU.
when I run torch I’m getting
GeForce RTX 3080 with CUDA capability sm_86 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 GeForce RTX 3080 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/
conda list:
_libgcc_mutex 0.1 conda_forge conda-forge
_openmp_mutex 4.5 1_gnu conda-forge
blas 1.0 mkl
bzip2 1.0.8 h7f98852_4 conda-forge
ca-certificates 2020.12.5 ha878542_0 conda-forge
certifi 2020.12.5 py38h578d9bd_1 conda-forge
cudatoolkit 11.1.1 h6406543_8 conda-forge
ffmpeg 4.3 hf484d3e_0 pytorch
freetype 2.10.4 h0708190_1 conda-forge
gmp 6.2.1 h58526e2_0 conda-forge
gnutls 3.6.13 h85f3911_1 conda-forge
intel-openmp 2020.2 254
jpeg 9b h024ee3a_2
lame 3.100 h7f98852_1001 conda-forge
lcms2 2.12 h3be6417_0
ld_impl_linux-64 2.33.1 h53a641e_7
libffi 3.3 he6710b0_2
libgcc-ng 9.3.0 h2828fa1_19 conda-forge
libgomp 9.3.0 h2828fa1_19 conda-forge
libiconv 1.16 h516909a_0 conda-forge
libpng 1.6.37 h21135ba_2 conda-forge
libstdcxx-ng 9.3.0 h6de172a_19 conda-forge
libtiff 4.1.0 h2733197_1
libuv 1.41.0 h7f98852_0 conda-forge
lz4-c 1.9.3 h9c3ff4c_0 conda-forge
mkl 2020.2 256
mkl-service 2.3.0 py38he904b0f_0
mkl_fft 1.3.0 py38h54f3939_0
mkl_random 1.1.1 py38h0573a6f_0
ncurses 6.2 he6710b0_1
nettle 3.6 he412f7d_0 conda-forge
ninja 1.10.2 h4bd325d_0 conda-forge
numpy 1.19.2 py38h54aff64_0
numpy-base 1.19.2 py38hfa32c7d_0
olefile 0.46 pyh9f0ad1d_1 conda-forge
openh264 2.1.1 h780b84a_0 conda-forge
openssl 1.1.1k h7f98852_0 conda-forge
pillow 8.2.0 py38he98fc37_0
pip 21.0.1 py38h06a4308_0
python 3.8.8 hdb3f193_5
python_abi 3.8 1_cp38 conda-forge
pytorch 1.8.1 py3.8_cuda11.1_cudnn8.0.5_0 pytorch
readline 8.1 h27cfd23_0
setuptools 52.0.0 py38h06a4308_0
six 1.15.0 py38h06a4308_0
sqlite 3.35.4 hdfb4753_0
tk 8.6.10 hbc83047_0
torchaudio 0.8.1 py38 pytorch
torchvision 0.9.1 py38_cu111 pytorch
typing_extensions 3.7.4.3 py_0 conda-forge
wheel 0.36.2 pyhd3eb1b0_0
xz 5.2.5 h7b6447c_0
zlib 1.2.11 h7b6447c_3
zstd 1.4.9 ha95c52a_0 conda-forge
nvidia-smi
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 455.32.00 Driver Version: 455.32.00 CUDA Version: 11.1 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|===============================+======================+======================|
| 0 GeForce RTX 3080 On | 00000000:65:00.0 On | N/A |
| 0% 54C P8 34W / 340W | 321MiB / 10014MiB | 10% Default |
| | | N/A |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=============================================================================|
| 0 N/A N/A 1987 G /usr/lib/xorg/Xorg 18MiB |
| 0 N/A N/A 2130 G /usr/bin/gnome-shell 64MiB |
| 0 N/A N/A 3878 G /usr/lib/xorg/Xorg 129MiB |
| 0 N/A N/A 4011 G /usr/bin/gnome-shell 87MiB |
| 0 N/A N/A 5999 G ...mviewer/tv_bin/TeamViewer 16MiB |
+-----------------------------------------------------------------------------+
can you please assist?