Ubuntu22.04下Anaconda与Pytorch环境搭建全攻略
1. Ubuntu22.04系统准备在开始安装Anaconda和Pytorch之前我们需要确保Ubuntu22.04系统已经做好充分准备。我建议先更新系统软件包这样可以避免后续安装过程中出现依赖问题。打开终端CtrlAltT执行以下命令sudo apt update sudo apt upgrade -y这个命令会更新软件包列表并升级所有可升级的软件包。根据网络速度不同这个过程可能需要几分钟到十几分钟不等。我在实际安装时发现有时候系统自带的Python版本可能与Anaconda产生冲突所以建议先检查系统Python版本python3 --versionUbuntu22.04默认会安装Python3.10这个版本与Anaconda和Pytorch都能很好地兼容。如果你之前安装过其他版本的Python或者Miniconda建议先彻底卸载它们避免环境冲突。我遇到过因为残留配置文件导致conda命令无法正常使用的情况清理方法如下sudo apt remove --purge python3-minimal python3 sudo rm -rf ~/.conda注意执行删除命令前请确认你真的需要移除这些组件特别是如果你有其他Python项目正在运行。2. Anaconda安装详解2.1 下载Anaconda安装包Anaconda的官方下载页面提供了多个版本我建议选择最新的稳定版。打开终端使用wget命令直接下载wget https://repo.anaconda.com/archive/Anaconda3-2023.09-0-Linux-x86_64.sh下载完成后先验证文件的完整性。这一步很多教程都会忽略但我强烈建议做因为网络传输过程中文件可能损坏sha256sum Anaconda3-2023.09-0-Linux-x86_64.sh将输出与官网提供的SHA256校验码对比确保完全一致再继续安装。如果校验失败重新下载即可。2.2 执行安装过程运行安装脚本时有几个关键点需要注意bash Anaconda3-2023.09-0-Linux-x86_64.sh安装过程中会提示几个重要选项按Enter键查看许可协议可以按空格快速翻页输入yes同意许可条款确认安装位置默认在用户目录下的anaconda3文件夹是否初始化conda选择yes我建议接受所有默认选项特别是最后一项初始化conda这会让conda命令自动添加到你的PATH环境变量中。安装完成后关闭并重新打开终端使环境变量生效。2.3 验证安装测试conda是否安装成功conda --version如果看到版本号输出如conda 23.9.0说明安装成功。你也可以启动Anaconda Navigator图形界面anaconda-navigator第一次启动可能会比较慢因为它需要初始化所有组件。如果遇到启动问题可以尝试更新condaconda update conda3. 创建和管理虚拟环境3.1 为什么要使用虚拟环境在深度学习项目中使用虚拟环境是必须的。我遇到过因为不同项目依赖库版本冲突导致的各种奇怪bug。虚拟环境可以隔离不同项目的依赖关系避免依赖地狱。Anaconda提供了非常方便的虚拟环境管理工具。3.2 创建Pytorch专用环境为Pytorch创建一个独立的环境是个好习惯。我建议使用Python3.8或3.9因为它们与Pytorch的兼容性最好conda create -n pytorch_env python3.8激活这个环境conda activate pytorch_env你会注意到终端提示符前面出现了(pytorch_env)表示当前处于这个虚拟环境中。要退出环境使用conda deactivate3.3 环境管理技巧列出所有虚拟环境conda env list删除不再需要的环境谨慎操作conda env remove -n pytorch_env克隆环境当你想基于现有环境创建新环境时很有用conda create --name new_env --clone pytorch_env4. Pytorch安装实战4.1 选择正确的Pytorch版本Pytorch官网提供了多种安装组合选择不当可能导致性能下降或兼容性问题。打开Pytorch官网(https://pytorch.org)根据你的硬件配置选择有NVIDIA显卡选择CUDA版本推荐11.8只有CPU选择CPU版本AMD显卡选择ROCm版本我建议即使有显卡也同时安装CPU版本因为有些操作在CPU上调试更方便。安装命令可以直接从官网复制但要注意以下几点确保conda环境已激活检查显卡驱动是否安装正确nvidia-smi命令如果网络不好可以添加清华镜像源4.2 实际安装过程对于大多数用户推荐使用conda安装自动处理依赖关系conda install pytorch torchvision torchaudio pytorch-cuda11.8 -c pytorch -c nvidia如果下载速度慢可以先配置清华镜像源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然后重新运行安装命令。安装完成后不要急着关闭终端因为第一次导入torch时会进行一些初始化工作可能需要几分钟。4.3 验证Pytorch安装启动Python解释器python在交互式环境中执行以下命令import torch print(torch.__version__) # 查看Pytorch版本 print(torch.cuda.is_available()) # 检查CUDA是否可用 torch.randn(3,3).cuda() # 测试GPU张量计算如果一切正常你应该能看到Pytorch版本号、True表示CUDA可用和一个3x3的随机矩阵。我遇到过cuda.is_available()返回False的情况通常是驱动版本不匹配导致的解决方法是重新安装对应版本的CUDA驱动。5. 常见问题与解决方案5.1 Conda命令找不到如果提示conda: command not found说明环境变量没配置正确。解决方法echo export PATH$HOME/anaconda3/bin:$PATH ~/.bashrc source ~/.bashrc5.2 Pytorch导入错误常见的导入错误包括libcudart.so找不到CUDA版本不匹配undefined symbol驱动版本太旧段错误显卡不兼容解决方法通常是重新安装匹配版本的CUDA驱动和Pytorch。可以先卸载再安装conda uninstall pytorch torchvision torchaudio conda install pytorch torchvision torchaudio pytorch-cuda11.8 -c pytorch -c nvidia5.3 虚拟环境激活失败如果conda activate失败可能是shell没有初始化conda。尝试source ~/anaconda3/bin/activate conda init bash然后重新打开终端。这个问题在新安装的系统中比较常见。5.4 磁盘空间不足Anaconda和Pytorch安装后可能占用几个GB空间。如果提示磁盘空间不足可以考虑安装Miniconda精简版Anaconda定期清理conda缓存conda clean --all安装时指定其他磁盘位置conda install -c pytorch pytorch --prefix/path/to/install6. 环境优化与配置6.1 配置Jupyter Notebook在虚拟环境中使用Jupyter Notebook可以方便地进行实验conda install jupyter jupyter notebook --generate-config为了让Jupyter识别conda环境需要安装ipykernelpython -m ipykernel install --user --namepytorch_env启动Notebook后就能选择pytorch_env内核了。我习惯配置Jupyter自动打开浏览器jupyter notebook --browserfirefox6.2 性能调优对于GPU用户有几个设置可以提升Pytorch性能torch.backends.cudnn.benchmark True # 启用cuDNN自动调优 torch.set_float32_matmul_precision(high) # 提高矩阵运算精度在训练循环前添加这些设置可以自动优化计算图。但要注意对于小模型可能反而会降低性能。6.3 环境导出与共享当项目需要协作时可以导出环境配置conda env export environment.yml其他人可以通过这个文件复现完全相同的环境conda env create -f environment.yml这个功能在团队协作和论文复现时特别有用。我通常会为每个项目单独创建一个environment.yml文件放在项目根目录下。
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2427006.html
如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!