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

1. nvidia-smi

复制代码

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

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

1. 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对照表

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。