wsl安装torch
时间: 2025-04-28 07:23:52 浏览: 28
### 如何在WSL中安装PyTorch
为了在 Windows Subsystem for Linux (WSL) 中成功安装 PyTorch,可以遵循如下方法:
#### 准备工作
确保已正确设置并运行 WSL2 版本的 Ubuntu 或其他支持的 Linux 发行版[^1]。
#### 更新包管理器索引文件
更新 APT 软件源列表以获取最新的软件包信息:
```bash
sudo apt update && sudo apt upgrade -y
```
#### 安装依赖项
安装 Python 和 pip 工具以及必要的开发工具链来编译某些扩展模块:
```bash
sudo apt install python3-pip build-essential libssl-dev libffi-dev python3-dev -y
```
#### 创建虚拟环境(可选)
建议创建一个新的 Python 虚拟环境来进行隔离化操作:
```bash
python3 -m venv pytorch-env
source pytorch-env/bin/activate
```
#### 使用 Conda 或 Pip 安装 PyTorch
可以选择通过 Anaconda 的 conda 命令或者直接利用 pip 来完成 PyTorch 及其依赖库的下载与配置。这里提供两种方式供选择:
##### 方法一:基于 Conda 的安装方案
对于希望简化管理和部署过程的情况,推荐采用 Miniconda 并执行以下命令:
```bash
# 下载 miniconda.sh 文件
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
# 运行脚本进行安装
bash Miniconda3-latest-Linux-x86_64.sh
# 关闭终端重新打开后再继续后续步骤...
conda create --name torch_env python=3.9
conda activate torch_env
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
```
##### 方法二:基于 Pip 的安装方案
如果不打算使用 Conda,则可以通过 pip 实现相同目的:
```bash
pip install torch torchvision torchaudio
```
注意,在此过程中可能还需要额外指定 CUDA 版本号等参数以便适配特定硬件条件下的 GPU 加速功能;具体细节请参阅官方文档中的指导说明。
#### 测试安装是否成功
最后验证一下 PyTorch 是否能够正常加载:
```python
import torch
print(torch.__version__)
if torch.cuda.is_available():
print('CUDA is available')
else:
print('No CUDA support found.')
```
阅读全文
相关推荐
















