Anaconda搭建深度学习虚拟环境

1、虚拟环境所需工具介绍

该虚拟环境搭建需要使用cuda+cudnn+pytorch+torchvision来实现。

组件 一句话
CUDA GPU 的"操作系统",让程序能指挥 GPU 并行计算
cuDNN 神经网络的"加速器",让深度学习运算更快
PyTorch 深度学习的"施工工具箱",用来搭建和训练模型
torchvision 计算机视觉的"标准件库",提供模型、数据集和图像工具

2、下载Anaconda

打开下面网址,按照下载程序下载,以下有两点注意其他按照默认即可

Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirrorhttps://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

第一点注意:

| Just Me | 仅安装到当前用户目录 | 不需要管理员权限,安装简单,卸载干净
| All Users | 安装到系统目录 | 需要管理员权限,适合多用户共用电脑

第二点注意:

这个直接选择跳过即可,这个界面是询问是否需要安装VSCode,建议使用Pycharm

3、虚拟环境操作指令

在搜索框搜素Anaconda Prompt,并打开输入下面指令,创建过程中会出现Proceed (y/n)?直接输入Y即可,创建完后即可进行激活

复制代码
1、创建虚拟环境
conda create -n 虚拟环境名 python=3.13(环境需要的python版本)

2、激活虚拟环境
conda activate 虚拟环境名

3、查看已创建的虚拟环境
conda env list

4、退出虚拟环境返回默认环境
conda deactivate

5、删除指定的虚拟环境(需先退出虚拟环境,-all参数确保删除该环境中的所有包和依赖项)
conda remove -n 虚拟环境名 --all

激活后则会自动进入虚拟环境,执行结果为下图所示:

4、安装PyTorch及CUDA

确定显卡支持的最高CUDA版本再点击下面的网址去官网下载支持该最高版本以下的PyTorch,使用下面的pip下载即可同时下载PyThorch和CUDA。具体操作看下图。

PyTorch官网下载链接https://pytorch.org/get-started/locally/第一步:桌面右键,选择显示更多选项,打开NVIDIA控制面板,点击左下角系统信息即可查看到最高支持CUDA版本

第二步:选择最高支持版本以下或相同的CUDA,复制下面链接再虚拟环境控制面板执行即可(需要打开创建好的虚拟环境进行下载)

复制代码
其中11.6为版本名,自己更换即可
pip install torch torchvision --index-url https://download.pytorch.org/whl/cu132

5、安装cuDNN

第一步:输入下面命令查看cudnn版本信息

复制代码
conda search cudnn --info

第二步:安装对应版本的cudnn,安装过程中弹出Proceed (y/n)?直接输入Y即可。

复制代码
后面的版本信息改为自己查看的版本即可
conda install cudnn=9.17.0.29

6、检验安装是否成功

输入下面代码进入python环境检验上面工具是否安装成功。(正确反馈为下图所示)

复制代码
python

1、验证pytorch是否安装成功
import torch

2、验证cuda是否安装成功
torch.cuda.is_available()
torch.version.cuda

3、验证cudnn是否安装成功
torch.backends.cudnn.is_available()
torch.backends.cudnn.version()
相关推荐
johnny2333 分钟前
Python生态调试库:IceCream、birdseye、peek、qj、crab_dbg、spewer
python
Cx330❀6 分钟前
【Linux网络】深入 HTTP 协议(五):从 Cookie/Session 原理到 C++ 源码实战
linux·运维·服务器·开发语言·网络·c++·http
鸿芯微控科技8 分钟前
MFC气体转换系数怎么用?标定气体、换算方向与Python批量计算
c++·python·mfc
北冥you鱼11 分钟前
Go语言四则运算实战:从基础类型到big包的深度解析
开发语言·后端·golang
江畔柳前堤8 小时前
roLabelImg 详细安装教程
开发语言·人工智能·后端·云原生
0566468 小时前
agent学习Day15——SQLAlchemy 查询过滤、分页与历史列表接口
python·学习·fastapi
Wang's Blog9 小时前
Go-Zero 项目开发47:自研微服务框架的必要性与核心结构设计
开发语言·微服务·golang
白鸽(二般)9 小时前
MinIO Java Client API
java·开发语言
hPw0eKIqD9 小时前
C++ 模板参数推导问题小记(非推导上下文)
开发语言·c++
程序员爱德华9 小时前
Python与C++:异同点对比
c++·python