[PYTHON] Use RTX 3090 with PyTorch

~~ * Information as of October 24, 2020. ~~ ** PyTorch 1.7 was released on October 28th, and it is compatible with CUDA 11 and can be used with RTX3090. ** **

Therefore, if you simply upgrade to PyTorch 1.7, you can use it with RTX3080 / 3090.

The following is old information.

Task

PyTorch 1.6 does not currently support RTX 3080/3090 and throws the following error.

GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation.

Solutions

The workaround is to install the development version of PyTorch 1.8.

pip install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cu110/torch_nightly.html

Although it is cu110, there is no problem because it works even if you are using cuda11.1.

version information

NVIDIA-SMI 455.23.05 Driver Version: 455.23.05 CUDA Version: 11.1/Ubuntu 20.04

result

I used to use 2070 SUPER, but the learning speed is about 3 times faster. Also, jax worked normally.

Ref.

Recommended Posts

Use RTX 3090 with PyTorch
Play with PyTorch
Cross-validation with PyTorch
Beginning with PyTorch
Use mecab-ipadic-neologd with igo-python
Use ansible with cygwin
Use pipdeptree with virtualenv
[Python] Use JSON with Python
Use Mock with pytest
Use indicator with pd.merge
Use Gentelella with django
Use mecab with Python3
Use tensorboard with Chainer
Use DynamoDB with Python
Use pip with MSYS2
Use Python 3.8 with Anaconda
Use pyright with Spacemacs
Story of trying to use tensorboard with pytorch
Use python with docker
Use TypeScript with django-compressor
Install torch-scatter with PyTorch 1.7
Use LESS with Django
Use MySQL with Django
Use Enums with SQLAlchemy
Use tensorboard with NNabla
Use GPS with Edison
Use nim with Jupyter
Use Trello API with python
Use shared memory with shared libraries
Use "$ in" operator with mongo-go-driver
Use custom tags with PyYAML
Use directional graphs with networkx
Use TensorFlow with Intellij IDEA
Use Twitter API with Python
Use pip with Jupyter Notebook
Try an autoencoder with Pytorch
Use DATE_FORMAT with SQLAlchemy filter
Use TUN / TAP with Python
Use sqlite3 with NAO (Pepper)
Try implementing XOR with PyTorch
Use sqlite load_extensions with Pyramid
Use Windows 10 fonts with WSL
Use chainer with Jetson TK1
Use SSL with Celery + Redis
Implement PyTorch + GPU with Docker
Use Cython with Jupyter Notebook
Prediction of Nikkei 225 with Pytorch 2
Machine learning Minesweeper with PyTorch
AWS Lambda with PyTorch [Lambda import]
Use Maxout + CNN with Pylearn2
Use WDC-433SU2M2 with Manjaro Linux
Use OpenBLAS with numpy, scipy
Prediction of Nikkei 225 with Pytorch
Perform Stratified Split with PyTorch
Use subsonic API with python3
I made Word2Vec with Pytorch
Use Sonicwall NetExtener with Systemd
Use prefetch_related conveniently with Django
Use AWS interpreter with Pycharm
Use Bokeh with IPython Notebook
Use Python-like range with Rust