torch torchvision version - Serbipunk/notes GitHub Wiki

official pip

https://download.pytorch.org/whl/torch_stable.html

check torch cuda details

torch.cuda.is_available()
torch.version.cuda

conda install

conda create --name pytorch170 python=3.7
conda activate pytorch170
conda install pytorch==1.7.0 torchvision torchaudio cudatoolkit=11.0 -c pytorch

conda create --name pytorch1110 python=3.8
conda activate pytorch1110
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio cudatoolkit=11.3 -c pytorch

cuda 11

https://www.zhaoyabo.com/?p=8291

https://download.pytorch.org/whl/torch_stable.html

cu117 supported ubuntu 22

cu117 torch [1.13.0 1.13.1] cu117 [0.14.0 0.14.1] py37 ~ py311

swin-beit

cu110 torch 1.7.1 torchvision 0.8.2

支持训练psp

cuda10

cu102 -f https://download.pytorch.org/whl/torch_stable.html torch==1.8.1+cu102 -f https://download.pytorch.org/whl/torch_stable.html torchvision==0.9.1+cu102 cu101 -f https://download.pytorch.org/whl/torch_stable.html torch==1.7.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html torchvision==0.8.2+cu101

conda env examples:

name: cc3090
channels:
- pytorch
- defaults
dependencies:
- python=3.8
- cudatoolkit=11.8  # real 11.7
- pytorch=1.13.0
- scipy
- pip
- pip:
  - dominate==2.6.0
  - torchvision==0.14.0
  - Pillow
  - numpy
  - visdom==0.1.8
  - wandb==0.12.18