栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

anaconda配置python环境_anaconda运行环境?

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

anaconda配置python环境_anaconda运行环境?

在线备忘录

(一)在windows系统中使用Anaconda配置深度学习环境

在线备忘录前言正文

1.查看NVIDIA显卡支持的CUDA驱动版本2.安装Anaconda3.建立新的虚拟环境4.安装Cudatoolkit,cuDNN及Pytorch 写在最后


前言

“在线备忘录”用于记录工作中遇到的部分技术问题,内容大多转载或整理自中文论坛博客,经个人二次加工形成,将尽量注明其直接出处。

由于时间、个人水平有限,且该备忘录主要目的为个人记录,因此该教程不一定适合没有接触过任命令行终端的新手,不详尽之处(some skips)烦请读者参考其他具体资料。


正文 1.查看NVIDIA显卡支持的CUDA驱动版本

打开NVIDIA控制面板(可搜索"NVIDIA Control Panel"),【系统信息】-【组件】,可看到NVIDIA驱动所支持的最高CUDA版本。


具体可参考该链接

如果还有游戏等用途,可以在NVIDIA官网下载CUDA,但是当只需要配置深度学习环境时使用anaconda安装pytorch/NVIDIA/…发布的cudatoolkit更方便,且可以切换不同版本(个人意见,欢迎指教)


2.安装Anaconda

(教程很多,此处略)


3.建立新的虚拟环境

打开Anaconda终端,在初始的base环境下创建虚拟环境。这里将虚拟环境名称设为test_env,python版本为3.8:

conda create -n test_env python=3.8

完成后显示:

查看所有环境,以检查虚拟环境是否已创建好:

conda env list

激活刚才建立的虚拟环境test_env,以在接下来使用pip/conda命令安装第三方库到该虚拟环境中(附:如果不使用wheel文件安装就不用关心其路径):

conda activate test_env

附:退出虚拟环境:

conda deactivate

附:删除虚拟环境:(conda remove -n [env_name] –all 参数为-n应该不完整)

conda remove --name [env_name] –all

4.安装Cudatoolkit,cuDNN及Pytorch

由于需要综合考虑python,pytorch,cudatoolkit与cuda驱动之间的关系,方便起见可以直接在torch或者anaconda官网上找torch库的安装命令。这里以torch官网的操作为例:

选好操作系统等参数后,点击button查看其他版本。
以torch1.8.0为例,使用该命令安装torch,将自动安装带有cudatoolkit(11.1),torchvision等组件:

将该命令复制到终端,执行:

(这里显示失败的原因可能与镜像失效有关吧,欢迎指教)

输入y以继续安装(当然倒回来检查一下路径和哪些依赖库总是一个好习惯)
1.2G,只好等几分钟吧)" />
(>1.2G,只好等几分钟吧)

之后还需安装cuDNN,否则此步骤完成后torch库的计算只能在CPU上运行,与GPU无关。

可到NVIDIA-cudnn找与cudatoolkit一致的cuDNN版本(需要注册),然后在anaconda官网上找该版本的cuDNN进行安装(从NVIDIA下载安装文件并配置比较麻烦)。

(Anaconda一般不需要注册,直接搜索第三方库名cudnn和版本号即可)


同样在终端运行以上命令,以安装cuDNN:

最后检查是否可用cuda加速计算:

法一:输入nvcc –version:

法二:依次执行:

python
import torch
torch.cuda.is_available()
torch.cuda.device_count()
exit()


环境配置成功。


写在最后

本过程在win10(64),anaconda on win32, conda = 4.12.0,cuda driver = 11.4亲测可用,linux系统除安装anaconda不一样以外其他步骤基本相同。
本文仅供参考。

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/783692.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号