win10 Ubuntu子系统 安装pytorch
时间: 2023-08-03 21:04:19 浏览: 178
Python安装Pytorch教程(图文详解).pdf
要在Windows 10的Ubuntu子系统上安装PyTorch,您可以按照以下步骤进行操作:
1. 打开Microsoft Store,并搜索Ubuntu。点击安装Ubuntu并按照提示进行安装。
2. 打开已安装的Ubuntu应用程序,完成初始设置和配置。
3. 在Ubuntu终端中,运行以下命令以更新软件包列表并安装依赖项:
```
sudo apt update
sudo apt upgrade
sudo apt install python3-pip python3-dev
```
4. 接下来,安装PyTorch。根据您的需求,可以选择使用CPU版本或者CUDA版本(如果您的机器具有NVIDIA GPU)。以下是安装命令:
- 安装CPU版本的PyTorch:
```
pip3 install torch torchvision
```
- 安装CUDA版本的PyTorch(要求您的机器具有NVIDIA GPU):
```
pip3 install torch torchvision torchaudio
```
注意:在安装CUDA版本的PyTorch之前,确保已经按照官方文档正确安装了NVIDIA驱动程序和CUDA工具包。
5. 安装过程可能需要一些时间,请耐心等待。一旦安装成功,您就可以在Ubuntu子系统中使用PyTorch了。
希望这可以帮助到您!如果您有任何其他问题,请随时提问。
阅读全文