当前位置:   article > 正文

pytorch1.7 安装_pytorch1.7下载

pytorch1.7下载

pip install torch==1.x.x -i https://pypi.douban.com/simple

pip install torchvision==0.x.x -i https://pypi.douban.com/simple

安装kornia

pip install kornia==0.x.x -i https://pypi.douban.com/simple

pytorch=1.3, kornia=0.3.1

pytorch=1.7, kornia=0.5.1

1.9

  1. # ROCM 4.2 (Linux only)
  2. pip install torch==1.9.0+rocm4.2 torchvision==0.10.0+rocm4.2 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
  3. # ROCM 4.1 (Linux only)
  4. pip install torch==1.9.0+rocm4.1 torchvision==0.10.0+rocm4.1 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
  5. # ROCM 4.0.1 (Linux only)
  6. pip install torch==1.9.0+rocm4.0.1 torchvision==0.10.0+rocm4.0.1 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
  7. # CUDA 11.1
  8. pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
  9. # CUDA 10.2
  10. pip install torch==1.9.0+cu102 torchvision==0.10.0+cu102 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
  11. # CPU only
  12. pip install torch==1.9.0+cpu torchvision==0.10.0+cpu torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

1.8.0

  1. # RocM 4.0.1 (Linux only)
  2. pip install torch -f https://download.pytorch.org/whl/rocm4.0.1/torch_stable.html
  3. pip install ninja
  4. pip install 'git+https://github.com/pytorch/vision.git@v0.9.0'
  5. # CUDA 11.1
  6. pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
  7. # CUDA 10.2
  8. pip install torch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0
  9. # CPU only
  10. pip install torch==1.8.0+cpu torchvision==0.9.0+cpu torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

1.7.1

  1. # CUDA 11.0
  2. pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
  3. # CUDA 10.2
  4. pip install torch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2
  5. # CUDA 10.1
  6. pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
  7. # CUDA 9.2
  8. pip install torch==1.7.1+cu92 torchvision==0.8.2+cu92 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
  9. # CPU only
  10. pip install torch==1.7.1+cpu torchvision==0.8.2+cpu torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

1.7.0

  1. # CUDA 11.0
  2. pip install torch==1.7.0+cu110 torchvision==0.8.0+cu110 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
  3. # CUDA 10.2
  4. pip install torch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0
  5. # CUDA 10.1
  6. pip install torch==1.7.0+cu101 torchvision==0.8.0+cu101 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
  7. # CUDA 9.2
  8. pip install torch==1.7.0+cu92 torchvision==0.8.0+cu92 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
  9. # CPU only
  10. pip install torch==1.7.0+cpu torchvision==0.8.0+cpu torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

1.6.0

  1. # CUDA 9.2
  2. conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=9.2 -c pytorch
  3. # CUDA 10.1
  4. conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.1 -c pytorch
  5. # CUDA 10.2
  6. conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.2 -c pytorch
  7. # CPU Only
  8. conda install pytorch==1.6.0 torchvision==0.7.0 cpuonly -c pytorch

1.5.0

  1. # CUDA 9.2
  2. conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=9.2 -c pytorch
  3. # CUDA 10.1
  4. conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.1 -c pytorch
  5. # CUDA 10.2
  6. conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.2 -c pytorch
  7. # CPU Only
  8. conda install pytorch==1.5.0 torchvision==0.6.0 cpuonly -c pytorch

1.4.0

  1. # CUDA 10.1
  2. pip install torch==1.4.0 torchvision==0.5.0
  3. # CUDA 9.2
  4. pip install torch==1.4.0+cu92 torchvision==0.5.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html
  5. # CPU only
  6. pip install torch==1.4.0+cpu torchvision==0.5.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

1.2.0

  1. # CUDA 10.0
  2. pip install torch==1.2.0 torchvision==0.4.0
  3. # CUDA 9.2
  4. pip install torch==1.2.0+cu92 torchvision==0.4.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html
  5. # CPU only
  6. pip install torch==1.2.0+cpu torchvision==0.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/盐析白兔/article/detail/193799
推荐阅读
相关标签
  

闽ICP备14008679号