RTX3090. python3.6 and pytorch1.10, pytorch1.7

ref:

• NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation
• 解决pytorch capability sm_86 is not compatible with the current PyTorch installation 问题

很麻煩的 torch. cuda.
用新的顯卡(其實也沒多新，就 3090而已)，使 sm85，需要某版本以上的 pytorch 才有support.
不然，雖然用 torch.cuda.is_available() 是 True, get_device_name() 也會正確顯示 RTX3090。
但是宣告一個在 GPU 的變數卻會有 Error:

>import torch
>print(torch.__version__)
1.10.2+cu102
>A = torch.rand(2000,128,device=torch.device('cuda'))
NVIDIA GeForce RTX 3090 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 NVIDIA GeForce RTX 3090 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

說其實 1.7.1 版的 pytorch 就已經 support RTX3090 了，是因為 pip repo 中的 pytorch package 配的 cuda 版本太舊，不 support RTX3090.
所以到 pytorch 網站下載新版 cuda 的 pytorch 來裝就可以了。

我的 機器 cuda 版本是 11.7，所以用：

pip install torch==1.10.0+cu113 torchvision==0.11.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

這樣裝完後，上面的測試command 就不會有 Error 了。

到 cu113 的 torch_stable 去看，支援只有從 torch 1.10.0 開始，所以可以用 ref(2) 的作法，到-f 的位址 cu110/torch_stable.html 去看裡面有沒有 torch-1.7.1
結果真的只有 cu110 有，cu111 之後都是 torch-1.10.0 了。

pip3 install torch==1.7.1+cu110 torchvision==0.8.2+cu110 -f https://download.pytorch.org/whl/cu110/torch_stable.html