python3.6搭建pytorch环境

---

前言

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。
• 检查驱动程序版本：

nvidia-smi

• 确保驱动程序版本 >= 450.80.02（CUDA 11.1 的最低要求）。

清华源安装命令 如果使用清华源（推荐），可以将官方 URL 替换为清华的 PyTorch 镜像：

pip install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple --extra-index-url https://download.pytorch.org/whl/cu111

---

资源

阿里云pytorch资源库
pytorch官网
pytorch对照表