文章目录
前言
PyTorch 1.8.1 是支持 Python 3.6 的最后一个版本。
CUDAToolkit 11.1 可以与 PyTorch 1.8.1 搭配使用（前提是你的 GPU 驱动程序支持 CUDA 11.1 或更高版本）。
一、国内镜像安装 PyTorch（支持 CUDA 11.1）
- 首先确保你已安装 NVIDIA 驱动程序，并且支持 CUDA 11.1。
- 检查驱动程序版本：
cpp
nvidia-smi- 确保驱动程序版本 >= 450.80.02（CUDA 11.1 的最低要求）。
清华源安装命令 如果使用清华源（推荐），可以将官方 URL 替换为清华的 PyTorch 镜像：
cpp
pip install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple --extra-index-url https://download.pytorch.org/whl/cu111