如何在conda虚拟环境中配置cuda+cudnn+pytorch深度学习环境
作者:江江ahh 发布时间:2022-06-16 14:43:04
首先,我们要明确,我们是要在虚拟环境中安装cuda和cuDNN!!!只需要在虚拟环境中安装就可以了。
下面的操作默认你安装好了python
一、conda创建并激活虚拟环境
前提:确定你安装好了anaconda并配置好了环境变量,如果没有,网上有很多详细的配置教程,请自行学习
在cmd命令提示符中输入conda命令查看anaconda
如果显示和上图相同,那么可以继续向下看
1.进入anaconda的base环境
方法1
在cmd命令提示符中输入如下命令
activate
方法2
直接在搜索栏里搜索anaconda prompt并打开即可
方法3
如果你是在linux操作系统下,在你的/anaconda/bin/下打开终端,即可直接进入anaconda的环境
以上三种方法出现(base)就意味着你已经进入了anaconda的基础环境
2.conda创建、进入和退出虚拟环境
这三个操作可以各用一行命令来完成
#创建一个虚拟环境
conda create -n [your_env_name(你的虚拟环境的名字)] python==[X.X](2.5、3.8等)
#eg:conda create -n nnunet_env python=3.8
#进入虚拟环境
conda activate [你的虚拟环境名]
#退出虚拟环境
conda deactivate
创建好的虚拟环境文件夹可以在anaconda文件夹中的envs文件夹里找到
后续的环境配置操作均要在激活虚拟环境的情况下完成!!!!!
二、查看CUDA版本
系统的CUDA版本,决定了系统最高可以支持什么版本的cudatoolkit,它是向下兼容的
我们可以通过nvidia-smi命令查看cuda版本号。
比如我的CUDA Version=11.7,那么它就可以支持 ≤ 11.7版本的CUDATookit
现在你已经知道了自己系统的cuda版本,接下来我会以自己利用清华镜像源来配置cuda+cudnn+pytorch深度学习环境的一个例子来简单直白的说明怎样去做
再提醒一遍,下面的操作也要在你激活自己的虚拟环境的情况下进行!
三、安装CUDATookit
我们需要用到下面这条命令
conda install cudatoolkit=11.3 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/
我选择下载11.3版本,这个需要看你们自己的需求来改变,只要你系统的CUDA支持就可以
四、安装cuDNN
如果你成功安装了你想要的那个版本的cudatookit,注意,现在你安装cudnn的版本必须依赖于cudatookit的版本
cuda与cudnn的对应关系可以在cuDNN历史版本下载页面看到:
cuda下载页面
cudnn下载页面
这里我简单列出来了较新的一些版本之间的对应关系
根据上图可知,我安装了CUDA 11.3版本,那么可选的cuDNN版本有很多,这里我直接无脑安装了最新版本,也就是cuDNN的8.4.0版本,同样是用清华镜像源来安装
conda install cudnn=8.4.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/
五、安装Pytorch
根据之前我们讲过的,pytorch的版本也是和CUDA版本有对应关系的,比如torch1.6.0只适配cuda10.2、10.1、9.2,不适配cuda11.0。
我们接下来的操作需要进入到Pytorch的官网Previous PyTorch Versions | PyTorch,在里面查看你想要的Pytorch版本它适配的CUDA版本并获取安装命令
比如我现在想要安装Pytorch的1.11.0版本,同时我之前已经安装了CUDA的11.3版本
如上图,这一条命令可以满足我的pytorch和cuda之间的对应关系,因此,我们复制它并运行,即可安装Pytorch 1.11.0
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch
注意:
也可以用pip install的命令来下载pytorch,但是因为三四两步都用的是conda,所以这里为了方便也用了condaconda install pytorch安装的是torch CPU版本,conda install pytorch torchvision -c pytorch安装的是GPU版本 检查环境是否配置成功
如果如下操作可以正常进行并打印出你安装的相应版本,那么你已经配置成功
#进入虚拟环境
conda activate [你的虚拟环境名]
#输入python来进入python的环境
python
#加载torch
import torch
print(torch.backends.cudnn.version())
#输出8200,代表着成功安装了cudnn v8.4.0
print(torch.__version__)
#输出1.11.0,代表成功安装了pytorch 1.11.0
print(torch.version.cuda)
#输出11.3,代表成功安装了cuda 11.3
torch.cuda.is_available()
#True
来源:https://blog.csdn.net/m0_68239345/article/details/128807483


猜你喜欢
- Python的版本是挺折腾人的,本着简单实用的原则我介绍一下我是如何安装多版本Python的。环境:windows10(64位)Python
- 在Apache, PHP, MySQL的体系架构中,MySQL对于性能的影响最大,也是关键的核心部分。对于Discuz!论坛程序也是如此,M
- 高效处理数据类型方法:处理数据In [1]: from random import randintIn [2]: data=[randint
- bootstrap中有alert组件,如果点击关闭按钮后该组件会被删除而不是被隐藏,想再显示怎么办呢?bootstrap-alert.js源
- 二维码作为一种信息传递的工具,在当今社会发挥了重要作用。从手机用户登录到手机支付,生活的各个角落都能看到二维码的存在。那你知道二维码是怎么解
- 1 引言 在关系数据库(DB2,Oracle,Sybase,Informix和SQL Server)最小的恢复和交易单位为一个事务(Tran
- 一、导入所需的库import turtleimport randomfrom math import *二、生成斐波那契数列斐波那契数列是指
- 一、首先进入官网https://www.python.org/1.点击Downloads 如图显示3.91 点击2.双击安装程序,进入下面的
- 1.彻底弄懂CSS盒子模式一(DIV布局快速入门) 2.彻底弄懂CSS盒子模式二(导航栏实例) 3.彻底弄懂CSS盒子模式三(浮动的表演和清
- 介绍百度aip模块是用于实现百度云与用户接口,简单来说就是使用百度云所拥有的人工智能模块。模块使用pip install baidu-aip
- 什么是性能分析?性能分析是衡量应用程序在代码级别的相对性能。性能分析将捕捉的事件包括:CPU的使用,内存的使用,函数的调用时长和次数,以及调
- 前言许多 Go 初学者在方法 * 的选择上可能会感到困惑,不知道该选择值 * 还是指针 * 。本文将会对方法 * 进行介
- 本文采用vue,同时增加鼠标点击事件和一些页面小优化基本结构新建一个sandBox.vue文件编写功能的基本结构 <div class
- 我在传递数据的时候老是出错,请问如何处理?不管是什么数据库,一般来说,是它所包含引号的字符串带来的问题。假设我们使用名为“strDate”的
- subplot函数介绍matplotlib下, 一个 Figure 对象可以包含多个子图(Axes), 可以使用 subplot() 快速绘
- virtualenv介绍virtualenv为应用提供了隔离的Python运行环境,解决了不同应用间多版本的冲突问题。virtualenv把
- 环境:Zend Studio 8.0Zend Studio是PHP开发者的首选开发工具,其地位相当于微软开发工具中的Visual Studi
- MySQL安全性指南(3) 作 者: 晏子2.4 不用GRANT设置用户如果你有一个早于3.22.11的MySQL版本,你不能使用GRANT
- 一个模块可以在逻辑上组织Python代码。将相关的代码到一个模块中,使代码更容易理解和使用。模块是可以绑定和借鉴任意命名属性的P
- 本文译文,采用意译。下面这些方法对于我来说很有作用,自从我发现了这些操作。1. 数组求和假设你有下面的数字数组:let numbers =