深度学习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())
相关推荐
wjykp1 分钟前
part6 PyTorch
人工智能·pytorch·python
liliangcsdn6 分钟前
DDIM扩散模型改进采样策略的推理探索
人工智能·深度学习·自然语言处理
读创商闻11 分钟前
中广融投让传统文化 “活” 起来
大数据·网络·人工智能
长桥夜波11 分钟前
【第二十五周】机器学习笔记
人工智能·笔记·机器学习
智驱力人工智能12 分钟前
在安全与尊严之间 特殊人员离岗检测系统的技术实现与伦理实践 高风险人员脱岗预警 人员离岗实时合规检测 监狱囚犯脱岗行为AI分析方案
人工智能·深度学习·opencv·算法·目标检测·cnn·边缘计算
Pith_12 分钟前
模式识别与机器学习复习笔记(上)
人工智能·笔记·机器学习
大任视点14 分钟前
云南首家现代农事综合服务中心在普洱思茅落户
大数据·人工智能
融云14 分钟前
融云 2025 回顾:「韧性」生长,「邪修」破局
人工智能·融云im·im选型
qdprobot15 分钟前
开源的在线串口调试助手支持macOS苹果电脑Windows系统Linux 浏览器webSerial
linux·运维·服务器·人工智能·mixly·小智ai·webserial