WSL环境安装CPU版本的pytorch和torchvision，要怎么做？

在WSL（Windows Subsystem for Linux）环境中安装CPU版本的PyTorch和 torchvision，你可以按照以下步骤操作： 1. **更新系统**: 首先，确保你的Ubuntu发行版已更新到最新状态，可以运行`sudo apt-get update && sudo apt-get upgrade`. 2. **安装依赖库**: 由于PyTorch需要一些额外的库支持，如gcc、cmake等，可以运行以下命令安装它们： ``` sudo apt-get install -y build-essential cmake libopenblas-dev luarocks libncurses5-dev libncursesw5-dev libreadline6-dev libdb5.3-dev libgdbm-dev liblzma-dev tk-dev libffi-dev python3-dev libssl-dev libsqlite3-dev libcurl4-openssl-dev libtbb-dev libjpeg-dev libpng-dev libwebp-dev libzmq3-dev ``` 3. **下载安装PyTorch**: 打开终端并导航到你想存放PyTorch的目录，然后从官方仓库下载CPU版本的PyTorch： ``` cd ~ wget https://download.pytorch.org/whl/cpu/torch-<version>-cpu.tar.gz tar xvf torch-<version>-cpu.tar.gz ``` 将 `<version>` 替换为你想要的PyTorch版本号。 4. **安装torchvision**: 安装完成后，进入刚解压的PyTorch文件夹，找到 `torch torchvision` 文件夹，同样下载对应版本的 torchvision whl 包并安装： ``` cd torch wget https://download.pytorch.org/whl/cpu/torchvision-<version>-cpu.tar.gz tar xvf torchvision-<version>-cpu.tar.gz ``` 然后将 `torchvision` 目录移动到 `python3.9/site-packages` 下，替换原来的文件： ``` mv torchvision-<version>/* /usr/local/lib/python3.9/dist-packages/ ``` 5. **验证安装**: 最后，检查安装是否成功，可以输入以下命令测试PyTorch和torchvision： ```bash python3 -c "import torch; print(torch.__version__)" python3 -c "from torchvision import transforms; print(transforms.__version__)" ```