如何在Windows WSL2（Ubuntu-20.04）安装支持GPU的PyTorch 0.3

报错根因 #

你遇到的wheel不支持报错，核心有两个原因：

1. PyTorch 0.3最高仅支持CUDA 9.0版本，WSL2当前默认的高版本CUDA环境和该版本不兼容
2. 你下载的wheel包适配ucs2编码的Python 3.6，部分conda环境默认安装的Python 3.6为ucs4编码，二者不匹配

操作步骤 #

1. 前置环境配置 #

• 首先在WSL2中安装CUDA 9.0，不要安装CUDA 10及以上版本
• 确认conda环境Python编码符合要求，执行命令：
  python -c "import sys; print(sys.maxunicode)"
  输出为65535即为符合要求的ucs2编码；如果输出为1114111，执行以下命令重装对应编码的Python 3.6：
  conda install python=3.6=*_ucs2
• 如果上述操作仍提示依赖冲突，直接新建干净的conda环境：

conda create -n pytorch03 python=3.6=*_ucs2
conda activate pytorch03

2. 执行安装 #

不需要手动下载wheel包，直接通过conda官方源执行安装命令即可：
conda install pytorch=0.3.0 cuda90 -c pytorch

3. 验证安装结果 #

进入Python交互环境执行以下命令：

import torch
print(torch.__version__)
print(torch.cuda.is_available())

输出0.3.0.post4和True即为GPU版PyTorch 0.3安装成功。

内容的提问来源于stack exchange，提问作者Arda Tümay