python3.6搭建pytorch环境
文章目录
- 前言
- 一、国内镜像安装 PyTorch（支持 CUDA 11.1）
- 资源
前言
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对照表