深度学习pytorch环境配置【保姆级】

程序员版本最好无论什么都用三年前的,比较保险

总步骤

1.下载nvidia显卡驱动 nvidia-smi验证

2.下载cuda nvcc --version 【cuda版本可以装多个,底层会识别】

3.下载miniconda 搜索anaconda prompt以管理员身份运行

4.anaconda基础命令创建环境

5.下载pytorch

6.创建pycharm项目,选择解释器

7.cudnn 提速 【可选,目前cuda下载自带应该】

目录

总步骤

1.下载英伟达驱动

[(1)nvidia-smi 检查驱动安装](#(1)nvidia-smi 检查驱动安装)

【可选】查询安装的CUDA版本

(2)【若未安装】驱动安装:

[2.下载cuda Toolkit11.8(长期稳定)](#2.下载cuda Toolkit11.8(长期稳定))

[(1)nvcc --version 确认cuda安装以及版本](#(1)nvcc --version 确认cuda安装以及版本)

[(2)【若未安装】下载cuda Toolkit11.8(长期稳定)](#(2)【若未安装】下载cuda Toolkit11.8(长期稳定))

点击步骤:

[最后选exe(local) 3Gb](#最后选exe(local) 3Gb)

不用管路径位置

推荐-》明白-》下一步-》不创建快捷方式

*验证安装需要重新打开黑窗口(管理员模式)

(3)在路径查看

3.下载miniconda

[(1)miniconda 各版本下载地址【空间更小】](#(1)miniconda 各版本下载地址【空间更小】)

[anaconda各版本下载地址: 【内容更全但不好管理】](#anaconda各版本下载地址: 【内容更全但不好管理】)

[(2)安装,选择all user](#(2)安装,选择all user)

位置在哪里都可以,这里e盘

默认勾选,安装

安装完成后,不用启动和欢迎直接finish

[*(3)验证环境anaconda prompt](#*(3)验证环境anaconda prompt)

4.anaconda基础命令创建环境

[(1)搜索anaconda prompt 以管理员身份运行!](#(1)搜索anaconda prompt 以管理员身份运行!)

(2)显示当前的conda环境列表

(3)创建新的conda环境并指定python版本

*如果遇到停止【可能是选中了,在键盘敲一个字母可能会好】

(4)切换conda环境

[(5)conda list当前环境有哪些包](#(5)conda list当前环境有哪些包)

(6)删除指定conda名字的环境

5.下载pytorch,镜像源+本地轮子

[(1)官方对照表:Previous PyTorch Versions](#(1)官方对照表:Previous PyTorch Versions)

(2)直接去网站下载安装包:download.pytorch.org/whl/torch/

(3)换成指定conda环境后,切换路径

(4)清华镜像源+本地torch-2.0.0.whl下载:

[conda list查看当前环境有哪些包](#conda list查看当前环境有哪些包)

6.创建pycharm项目,选择解释器

(1)添加新的解释器+添加本地解释器

(2)选择现有,选择Conda,选择conda.bat

(3)测试代码,测试torch

[7.cudnn 提速30%](#7.cudnn 提速30%)


1.下载英伟达驱动

(1)nvidia-smi 检查驱动安装

查看支持的最高CUDA版本()不代表已安装

nvidia-smi

最多安装cuda 12.6版本【常用的用11.8】【对应pytorch2.0最好】

【可选】查询安装的CUDA版本

CUDA Toolkit 13.1 - 发布说明 --- 发布说明 13.1 文档

已安装驱动:你会看到一个包含GPU信息的表格界面

(2)【若未安装】驱动安装:

NVIDIA GeForce 驱动程序 - N 卡驱动 | NVIDIA

2.下载cuda Toolkit11.8(长期稳定)

(1)nvcc --version 确认cuda安装以及版本

nvcc --version

(2)【若未安装】下载cuda Toolkit11.8(长期稳定)

CUDA Toolkit Archive | NVIDIA Developer

点击步骤:

步骤:downloads-》cuda toolkit(learn more) -》download Now -》archive...release -》cuda toolkit 11.8.0

【cuda版本可以装多个,底层会切换】

最后选exe(local) 3Gb

CUDA Toolkit 11.8 Downloads | NVIDIA Developer

不用管路径位置

推荐-》明白-》下一步-》不创建快捷方式

*验证安装需要重新打开黑窗口(管理员模式)

(3)在路径查看

3.下载miniconda

【免费】Miniconda3-py38-22.11.1-1-Windows-x86-64.zip资源-CSDN下载

(1)miniconda 各版本下载地址【空间更小】

官网:Miniconda - Anaconda 点击overview,地址 https://repo.anaconda.com/miniconda.

miniconda 往期版本 选择三年前最好

base python版本3.8 miniconda版本22.11.1-1

anaconda各版本下载地址: 【内容更全但不好管理】

Index of /archive

(2)安装,选择all user

位置在哪里都可以,这里e盘

默认勾选,安装

允许全局访问

base环境默认不用装任何东西,但是默认勾选以后可能有其他用途

安装完成后,不用启动和欢迎直接finish

*(3)验证环境anaconda prompt

搜索anaconda prompt 以管理员身份运行!

4.anaconda基础命令创建环境

(1)搜索anaconda prompt 以管理员身份运行!

(2)显示当前的conda环境列表

conda env list

目前只有base一个环境,盒子

(3)创建新的conda环境并指定python版本

conda create -n <name> python=3.8

示例:

bash 复制代码
conda create -n torch_test python=3.8

<name> 起自己的环境盒子名字

*如果遇到停止【可能是选中了,在键盘敲一个字母可能会好】

(4)切换conda环境

conda activate <name>

<name> 起自己的环境盒子名字, torch_test

示例:

bash 复制代码
conda activate torch_test

(5)conda list当前环境有哪些包

conda list

(6)删除指定conda名字的环境

conda remove -n <name>--all

pytorch_test01

删除后:

5.下载pytorch使用镜像

(1)官方对照表:Previous PyTorch Versions

(2)直接去网站下载安装包:download.pytorch.org/whl/torch/

夸克 :https://pan.quark.cn/s/0e953ef0cd8e

依次点开torch、torchvision与torchaudio三个库,分别crtl+F,直接搜索库对应的版本号+cu118

(3)换成指定conda环境后,切换路径

cd /d E:\conda\miniconda\Localfromlibinstall\torch

(4)清华镜像源+【local】torch-2.0.0.whl下载:

输入pip install,然后按tab键就会识别补全

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple "torch-2.0.0+cu118-cp38-cp38-win_amd64.whl"

conda list查看当前环境有哪些包

conda list

6.创建pycharm项目,选择解释器

(1)添加新的解释器+添加本地解释器

(2)选择现有,选择Conda,选择conda.bat

选择之前创建的环境 torch_test

(3)测试代码,测试torch

python 复制代码
import torch

# 验证CUDA是否可用
print(torch.cuda.is_available())

# 打印torch版本
print(torch.__version__)

# 打印CUDA版本
print(torch.version.cuda)

# 打印cuDNN版本
# print(torch.backends.cudnn.version())

7.cudnn 硬件提速30%

目前下载cuda好像自带

cpp 复制代码
# 打印cuDNN版本
print(torch.backends.cudnn.version())

cuDNN Archive | NVIDIA Developer

csdn库:【免费】cudnn-windows-x86-64-8.6.0.163-cuda11.zip资源-CSDN下载

cpp 复制代码
import torch

# 验证CUDA是否可用
print(torch.cuda.is_available())

# 打印torch版本
print(torch.__version__)

# 打印CUDA版本
print(torch.version.cuda)

# 打印cuDNN版本
print(torch.backends.cudnn.version())
相关推荐
明王明王23 分钟前
从零搭建一个 AI Infra 实验室㉑:llm-d——从单个 vLLM 走向分布式推理
人工智能·ai infra
武子康25 分钟前
图片删了,为什么旧链接还能打开、后台还能生成回来?
人工智能·agent
yuanyuan3341 小时前
2026枣庄带检测报告伸缩门品牌测评|本地服务
大数据·数据库·人工智能·伸缩门
无所畏俱的老尹1 小时前
Agent 落地为什么需要影子对照:同输入只读双跑、写动作变提案、差异报警,比直接切全量更安全
人工智能·机器学习·云计算·aigc·devops
神鹰手哥_起飞版2 小时前
深度研判任务里,开思考、换更强的模型、套方法论,哪个更有用?
人工智能
ObserverX212 小时前
我是如何用 13KB 纯 Python,发现本地 RAG 的“隐式递归记忆”的
人工智能
Jason_gao9992 小时前
Github Copilot 新手极速上手指南
人工智能·github·copilot
用户360055579002 小时前
Kestrel_VLLM - 35 · 把 m 循环搬进内核:一次「先量发射、再合并发射」的完整回合
人工智能
Linye2 小时前
为什么让 AI 修个小 Bug 却把底层全重写了?聊聊 Coding Agent 的最小修改边界
人工智能·架构
深度开源2 小时前
我扫了自己电脑里的 402 个 AI 技能,光目录就吃掉 4.5 万 token
人工智能