python深度学习环境配置扫盲

ubuntu系统下,基于gpu和python做深度学习,需要安装的内容及作用:

1.显卡驱动:调用显卡的基础,安装成功后,可以调用nvidia-smi指令,查看显卡驱动版本

2.cuda和cudnn:nvidia为显卡加速计算提供的库,cuda和显卡驱动是有对应关系的

为了便于多个不同代码的调试,我们可能需要多个版本的cuda,只要将多个cuda都解压到/usr/local目录即可,具体使用哪个,可以通过用户的环境变量配置,使不同的文件夹(用户/环境)使用不同的cuda版本。

用户环境变量修改如下:vim ~/.bashrc

找到其中的cuda的定义,更改指向路径,即可更改调用的cuda版本

修改完还后,运行nvcc -V指令,可以查看当前cuda的版本

关闭终端,新开终端可生效

3.anaconda:python的环境管理工具,通过anaconda管理多个版本的python环境很方便,比直接安装python和pip要清晰清楚。

anaconda多环境管理

conda env list指令可查看所有环境

conda create -n name python=3.6可建立新的环境

conda activate name可激活指定环境

conda install xxx安装指定的包

tensorflow,pytorch等架构,涉及python版本/cuda版本兼容问题,可从下面网址查看有哪些可用的版本,再配置环境,环境配置错误,则无法找到目标文件,无法安装。

查看清华源中版本对应关系的网址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

比如:安装pytorch时,可使用如下指令

conda install pytorch=1.0.1 torchvision cudatoolkit=8.0


但如果使用

conda install pytorch=1.0.1 torchvision cudatoolkit=9.0

则无法安装,因为清华的源中,没有用cuda9.0编译的pytorch

tensorflow本身和cuda版本存在对应关系,如下图:

 

4.pycharm:集成开发环境,查阅代码方便。

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值