pytorch-cuda版本匹配
安装pytorch
pip install torch==1.1.0
pip install torch==0.4.0
pip install torch==1.0.0
pip install torch==1.5.0
pip install torch==1.4.0
查看pytorch是否支持CUDA
torch.cuda.is_available()
pytorch-cuda版本对应
- torch 1.1.0 -> CUDA 9.2
- torch 1.2.0 -> CUDA 10.0
- torch 1.3.0 -> CUDA 10.1
- torch 1.4.0 -> CUDA 10.2
- torch 1.5.0 -> CUDA 10.2