文章目录
前言
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