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