python3.6搭建pytorch环境

文章目录

  • 前言
  • [一、国内镜像安装 PyTorch(支持 CUDA 11.1)](#一、国内镜像安装 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。
  • 检查驱动程序版本:
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

资源

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

相关推荐
阿_旭4 分钟前
基于深度学习的蛀牙智能检测与语音提示系统【python源码+Pyqt5界面+数据集+训练代码】
人工智能·python·深度学习·毕业设计·蛀牙检测
Phodal7 分钟前
可编排 AI 编程助手 Shire 1.0 发布:一键连接工具生态,重塑软件开发流程
人工智能·ai 编程
iisugar11 分钟前
AI第一天 自我理解笔记--超参数
人工智能·笔记
冷琴199620 分钟前
基于python+django+vue.js开发的社区养老管理系统源码+运行步骤
vue.js·python·django
朱剑君23 分钟前
人工智能与机器学习——系统学习规划
人工智能·学习·机器学习
AI服务老曹1 小时前
以实现生产制造、科技研发、人居生活等一种或多种复合功能的智慧油站开源了
人工智能·科技·自动化·音视频·生活·制造
风清扬雨1 小时前
计算机视觉中的MIP算法全解析
人工智能·算法·计算机视觉
zsmydz8881 小时前
手机遥控开关技术解析与应用指南
人工智能·生活·智能硬件
掘金安东尼1 小时前
🚀使用 Trae + SSH “急速开发”一个可以公网访问的 web 社区应用!!
人工智能·面试·llm