从零开始搭建深度学习环境

本文详细介绍了如何在Ubuntu18.04上从零开始搭建深度学习环境,包括安装基础环境、显卡驱动、CUDA10.0、cuDNN7.4.2以及TensorFlow-GPU。步骤清晰,涵盖了从安装Anaconda3到配置国内镜像、创建TensorFlow环境并安装相关库的过程。

开始深度学习调参之路不能没有一个趁手的环境

安装基础环境

我使用的机器环境是:Ubuntu18.04

  1. 安装显卡驱动
    1.
    2. 按照cuda所需的版本安装
    3. sudo ubuntu-drivers autoinstall
  2. 安装pip
    1. sudo apt-get install python-pip
  3. 下载anaconda3
    1. 地址:https://www.anaconda.com/distribution/
    2. 安装路径:/home/anaconda3
    3. 进入下载路径:bash Anaconda3-5.2.0-Linux-x86_64.sh
    4. 打开.bashrc验证anaconda环境变量
  4. 查看NVIDIA图形卡和推荐的驱动程序的模型:ubuntu-drivers devices
    1.
  5. 验证显卡驱动版本: cat /proc/driver/nvidia/version
  6. 安装cuda
    1. 版本选择:
      1. 只有cudnn7.4.2支持Ubuntu18.04和16.04
      2. cuDnn 7.4.2需要cuda10.0, cuda10.0需要gcc版本是7.3.0,而Ubuntu18.04的GCC版本正是7.3,无需降级,因此选择cuda版本为10.0
    2. 下载cuda10.0:
<think>嗯,用户想从零开始搭建深度学习环境,包括安装框架和配置依赖。首先,我需要整理出详细的步骤,确保覆盖所有必要环节。用户可能对Anaconda和Pyycharm不太熟悉,所以得先解释这两个工具的作用。 根据引用内容,Anaconda用来管理环境,安装不同版本的Python和各种库,而PyCharm是IDE,方便编写代码。用户提到国内镜像源,可能下载速度是个问题,所以配置镜像源是必要的步骤。引用[1]和[2]都提到了Anaconda的环境创建和镜像配置,这部分需要详细说明。 接下来是安装深度学习框架,比如TensorFlow、Keras、PyTorch。用户可能不确定该装哪个版本,特别是CUDA和cuDNN的兼容性。引用[3]提到NVIDIA驱动的问题,需要提醒用户检查显卡驱动,安装正确的CUDA版本。比如PyTorch官网会根据CUDA版本提供安装命令,这点很重要。 然后,配置PyCharm的环境,确保能正确使用Anaconda创建的环境。用户可能遇到路径设置的问题,需要指导如何在PyCharm中选择解释器。 另外,用户可能忽略环境隔离的重要性,所以需要强调为每个项目创建独立环境的好处,避免版本冲突。引用[2]里提到创建Python3.9的环境,可能需要解释为什么推荐这个版本,或者根据框架需求选择。 测试部分,需要简单代码示例验证框架是否安装成功,比如输出TensorFlow或PyTorch的版本,以及检查GPU是否可用。这能帮助用户确认安装是否正确。 最后,常见问题解答,比如镜像源配置失败、CUDA不可用,这些是用户可能遇到的典型问题,需要给出解决方案,比如检查驱动版本,重新安装CUDA工具包等。 总结步骤应该是:安装Anaconda→配置镜像源→创建环境→安装框架→配置PyCharm→测试环境。每个步骤都要详细说明命令和注意事项,确保用户能一步步跟着操作,避免遗漏关键点。</think>### 从零搭建深度学习环境的详细步骤(以PyTorch/TensorFlow为例) #### 一、基础环境准备 1. **安装Anaconda** - 作用:管理Python环境和依赖包隔离 - 下载地址:https://www.anaconda.com/download - 验证安装: ```bash conda --version ``` 2. **配置国内镜像源**(加速下载) ```bash conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ conda config --set show_channel_urls yes ``` #### 二、创建独立环境 1. **新建Python3.9环境**(推荐版本) ```bash conda create -n dl_env python=3.9 # dl_env为环境名 conda activate dl_env ``` 2. **安装基础工具包** ```bash conda install numpy pandas matplotlib jupyter ``` #### 三、GPU环境配置(需NVIDIA显卡) 1. **检查驱动兼容性** - 执行`nvidia-smi`查看驱动版本 - 根据驱动版本选择CUDA版本(参考NVIDIA官网兼容表) 2. **安装PyTorch环境** ```bash # 访问https://pytorch.org获取最新安装命令 conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia ``` 3. **安装TensorFlow环境** ```bash pip install tensorflow[and-cuda] # 自动匹配CUDA版本 ``` #### 四、IDE配置(PyCharm) 1. **创建新项目时选择Existing interpreter** 2. 定位到Anaconda安装路径下的`envs/dl_env/python.exe` 3. 验证环境: ```python import torch print(torch.__version__, torch.cuda.is_available()) # 应输出True[^2] ``` #### 五、验证安装 ```python # PyTorch测试 import torch x = torch.rand(5,3) print(x) # TensorFlow测试 import tensorflow as tf print(tf.reduce_sum(tf.random.normal([1000, 1000]))) ``` ### 常见问题解决方案 1. **CUDA不可用** - 检查驱动版本与CUDA版本匹配 - 重装CUDA Toolkit:https://developer.nvidia.com/cuda-toolkit 2. **包下载失败** - 重置镜像源:`conda config --remove-key channels` - 尝试pip安装:`pip install -i https://pypi.tuna.tsinghua.edu.cn/simple 包名` 3. **环境路径冲突** - 使用`conda clean --all`清理缓存 - 避免在系统路径安装包 [^1]: 环境隔离是保证项目依赖纯净的关键 [^2]: PyTorch官网提供自动匹配CUDA版本的安装命令 [^3]: 显卡驱动需通过NVIDIA控制面板验证
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值