PyTorch和CUDA的安装方法

PyTorch可支持的CUDA版本以及对应的安装命令

Start Locally | PyTorchAn open source machine learning framework that accelerates the path from research prototyping to production deployment.https://pytorch.org/get-started/locally/

CUDA可下载包

CUDA Toolkit Archive | NVIDIA DeveloperPrevious releases of the CUDA Toolkit, GPU Computing SDK, documentation and developer drivers can be found using the links below. Please select the release you want from the list below, and be sure to check www.nvidia.com/drivers for more recent production drivers appropriate for your hardware configuration.https://developer.nvidia.com/cuda-toolkit-archive

 先安装CUDA然后再用PyTorch命令安装torch（不知道反过来能否成功）

检测CUDA当前版本号

nvcc -V

检测是否已成功安装对应的Torch和CUDA

import torch
print("torch.cuda.is_available() =", torch.cuda.is_available())

结果应该是:

torch.cuda.is_available() = TRUE