VMware虚拟机中体验PyTorch：Ubuntu深度学习环境搭建备用方案

VMware虚拟机中体验PyTorchUbuntu深度学习环境搭建备用方案1. 为什么选择虚拟机方案深度学习环境搭建对很多初学者来说是个头疼的问题。直接在物理机上安装可能会影响现有系统而云服务器又需要持续付费。这时候VMware虚拟机提供了一个完美的中间方案——既能隔离环境又能免费使用。用虚拟机跑PyTorch主要有这些好处系统隔离不会影响你的主操作系统环境纯净可以随时创建、删除、重置成本低廉相比云服务器本地虚拟机几乎零成本学习友好适合反复折腾、测试不同配置当然虚拟机性能肯定比不上物理机或专业云服务器。但对于学习PyTorch基础、跑小规模模型来说完全够用了。特别是当你只是想先体验一下或者测试一些简单代码时这个方案特别合适。2. 准备工作与环境搭建2.1 硬件与软件需求在开始之前先确认你的电脑满足这些基本要求主机配置建议至少16GB内存固态硬盘多核CPUi5及以上VMware软件推荐使用VMware Workstation Pro16或17版Ubuntu镜像下载Ubuntu 20.04 LTS或22.04 LTS版本存储空间建议给虚拟机分配至少50GB空间如果你的主机有NVIDIA显卡并且想尝试GPU加速还需要确认显卡支持CUDA查NVIDIA官网下载对应的显卡驱动2.2 创建Ubuntu虚拟机现在我们来一步步创建虚拟机打开VMware点击创建新的虚拟机选择自定义(高级)安装类型硬件兼容性选择最新版本如Workstation 16.x选择稍后安装操作系统客户机操作系统选择Linux版本选Ubuntu 64位给虚拟机起个名字比如PyTorch-Ubuntu处理器配置建议给2-4个内核根据你主机CPU情况内存至少分配8GB如果主机有32GB可以给16GB网络类型选择NAT最简单I/O控制器类型保持默认磁盘类型SCSI选择创建新虚拟磁盘磁盘大小建议50GB选择将虚拟磁盘拆分成多个文件指定磁盘文件位置选个剩余空间大的盘完成创建创建完成后还需要进行一些关键设置点击编辑虚拟机设置在CD/DVD选项中选择你下载的Ubuntu ISO镜像如果要用GPU在显示器中开启3D加速点击确定保存设置3. 安装Ubuntu系统现在可以启动虚拟机安装Ubuntu了点击开启此虚拟机选择Install Ubuntu语言选择English或中文简体键盘布局保持默认安装类型选择正常安装勾选安装Ubuntu时下载更新磁盘分区选择清除整个磁盘并安装Ubuntu放心这只会影响虚拟机设置你的位置影响时区创建用户账户记住你设的用户名和密码等待安装完成约15-30分钟取决于你的主机性能安装完成后重启第一次进入系统后建议先做这些基础设置运行软件更新sudo apt update sudo apt upgrade -y安装VMware Tools增强工具在VMware菜单选择虚拟机→安装VMware Tools在Ubuntu中会自动挂载CD打开终端运行安装脚本调整屏幕分辨率如果显示不正常4. 安装PyTorch环境4.1 基础依赖安装首先安装一些必要的工具和库sudo apt install -y python3-pip python3-dev git curl wget sudo apt install -y build-essential cmake建议创建一个专门的Python虚拟环境sudo apt install -y python3-venv python3 -m venv pytorch_env source pytorch_env/bin/activate4.2 安装PyTorch根据你的虚拟机配置选择适合的PyTorch版本CPU版本推荐大多数用户pip3 install torch torchvision torchaudioGPU版本仅当配置了GPU直通 先安装CUDA工具包和cuDNN然后安装对应版本的PyTorch去PyTorch官网查最新命令验证安装是否成功python3 -c import torch; print(torch.__version__)4.3 安装常用数据科学库为了完整的数据科学环境建议安装这些常用库pip3 install numpy pandas matplotlib scikit-learn jupyter notebook安装Jupyter后可以这样启动jupyter notebook --ip0.0.0.0 --no-browser然后在主机浏览器访问虚拟机IP加上8888端口如http://192.168.x.x:88885. 性能优化与配置5.1 虚拟机性能调优为了让虚拟机运行更流畅可以尝试这些优化分配更多资源在虚拟机设置中增加CPU核心数和内存启用3D加速在显示器设置中勾选加速3D图形调整优先级在主机任务管理器中给VMware进程设置高优先级关闭特效在Ubuntu设置中关闭动画和视觉效果使用SSD如果主机有SSD把虚拟机文件放在SSD上5.2 网络配置技巧默认NAT网络可能速度较慢可以尝试桥接模式让虚拟机获得独立IP像物理机一样接入网络端口转发把虚拟机的服务端口映射到主机共享文件夹设置主机和虚拟机之间的文件共享6. 虚拟机方案与云服务器对比最后我们来对比一下虚拟机方案和直接使用GPU云服务器的优缺点对比项VMware虚拟机GPU云服务器成本一次性投入VMware软件按小时/月付费性能受限于主机硬件专业级GPU性能灵活性随时创建/删除需要申请/释放隔离性完全隔离共享环境适用场景学习/测试/小模型大规模训练/生产虚拟机适合初学者学习PyTorch基础测试代码和小规模实验需要完全控制的环境预算有限的个人用户云服务器适合需要强大GPU算力运行大规模模型训练团队协作开发生产环境部署获取更多AI镜像想探索更多AI镜像和应用场景访问 CSDN星图镜像广场提供丰富的预置镜像覆盖大模型推理、图像生成、视频生成、模型微调等多个领域支持一键部署。