vmware ubuntu 安装pytorch
时间: 2025-05-02 15:48:20 浏览: 30
### 在 VMware 虚拟机中的 Ubuntu 系统上安装 PyTorch
要在 VMware 虚拟机上的 Ubuntu 系统中成功安装 PyTorch,可以按照以下方法操作:
#### 1. 更新系统包
在开始之前,确保系统的软件包是最新的。打开终端并执行以下命令来更新和升级现有的软件包:
```bash
sudo apt update && sudo apt upgrade -y
```
#### 2. 安装 Python 和 pip
如果尚未安装 Python 或者 `pip` 工具,则需要先完成这一步骤。可以通过以下命令安装 Python3 及其依赖工具:
```bash
sudo apt-get install python3 python3-pip -y
```
验证已安装的 Python 版本是否满足需求[^2]:
```bash
python3 --version
```
#### 3. 创建虚拟环境 (可选)
为了更好地管理项目依赖关系,建议创建一个新的虚拟环境。这里以 Conda 虚拟环境为例说明如何设置。
##### 使用 Anaconda/Miniconda 创建虚拟环境
通过以下命令创建一个名为 `pytorch_env` 的新虚拟环境,并指定 Python 版本为 3.9[^3]:
```bash
conda create -n pytorch_env python=3.9
```
激活该虚拟环境:
```bash
conda activate pytorch_env
```
#### 4. 安装 PyTorch
访问官方 PyTorch 官网获取适合当前硬件配置的最佳安装指令:https://pytorch.org/get-started/locally/
通常情况下,默认推荐 CUDA 支持版本适用于 NVIDIA GPU 用户;对于无显卡支持或者仅需 CPU 计算的情况可以选择纯 CPU 版本。以下是针对不同场景的具体安装方式:
##### 如果有 NVIDIA 显卡且希望利用 CUDA 加速计算:
假设已经正确设置了驱动程序以及 cuDNN 库,那么可以直接运行如下命令下载对应版本的 PyTorch 包含 CUDA 支持功能:
```bash
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117
```
##### 若只需基于 CPU 的运算能力即可满足应用需求的话:
则应采用下面这条简化形式来进行部署:
```bash
pip install torch torchvision torchaudio
```
#### 5. 验证安装结果
最后,在同一会话窗口里测试一下刚刚加载好的库能否正常工作。尝试导入模块并打印版本号作为初步确认手段之一:
```python
import torch
print(torch.__version__)
if torch.cuda.is_available():
print('CUDA is available')
else:
print('Only using CPU.')
```
以上就是在 VMware 上搭建好 Ubuntu 后进一步实现 PyTorch 开发环境构建的过程概述[^1]^.
阅读全文
相关推荐

















