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

相关推荐
FL1623863129几秒前
使用yolov8+flask实现精美登录界面+图片视频摄像头检测系统
python·yolo·flask
阿木实验室1 分钟前
论文分享 | HE-Nav: 一种适用于复杂环境中空地机器人的高性能高效导航系统
人工智能·机器人·开源·无人机·空地机器人
计算机软件程序设计16 分钟前
Django中的查询条件封装总结
后端·python·django
咖啡调调。18 分钟前
Django连接MySQL
python·mysql·django
程序员Linc19 分钟前
NPU、边缘计算与算力都是什么啊?
人工智能·边缘计算
星哥说事35 分钟前
开源通义万相本地部署方案,文生视频、图生视频、视频生成大模型,支持消费级显卡!
人工智能·开源·音视频
来自于狂人37 分钟前
当大模型训练遇上“双向飙车”:DeepSeek开源周 DualPipe解析指南
人工智能·算法·系统架构·gpu算力
搏博40 分钟前
人工智能的数学基础之概率论与统计学(含示例)
图像处理·人工智能·机器学习·数据分析·概率论
栀子清茶1 小时前
Towards Universal Soccer Video Understanding——论文学习(足球类)
论文阅读·人工智能·深度学习·学习·算法·计算机视觉·论文笔记
木易:_/1 小时前
【004】deepseek本地化部署后,python的调用方式_#py
开发语言·python