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

相关推荐
fakaifa18 分钟前
【最新版】CRMEB Pro版v3.4系统源码全开源+PC端+uniapp前端+搭建教程
人工智能·小程序·uni-app·php·crmeb·源码下载·crmebpro
TuringAcademy4 小时前
AAAI爆款:目标检测新范式,模块化设计封神之作
论文阅读·人工智能·目标检测·论文笔记
The Open Group7 小时前
英特尔公司Darren Pulsipher 博士:以架构之力推动政府数字化转型
大数据·人工智能·架构
喂完待续7 小时前
【Tech Arch】Spark为何成为大数据引擎之王
大数据·hadoop·python·数据分析·spark·apache·mapreduce
Ronin-Lotus7 小时前
深度学习篇---卷积核的权重
人工智能·深度学习
.银河系.7 小时前
8.18 机器学习-决策树(1)
人工智能·决策树·机器学习
敬往事一杯酒哈7 小时前
第7节 神经网络
人工智能·深度学习·神经网络
三掌柜6667 小时前
NVIDIA 技术沙龙探秘:聚焦 Physical AI 专场前沿技术
大数据·人工智能
2502_927161288 小时前
DAY 42 Grad-CAM与Hook函数
人工智能
Hello123网站8 小时前
Flowith-节点式GPT-4 驱动的AI生产力工具
人工智能·ai工具