众所周知,jittor对windows的支持是一言难尽,各种cuda问题层出不穷。
为了在windows上使用jittor,需要在wsl中安装jittor。

安装 WSL : ubuntu22LTS

  1. 启用hyper-v
  2. 安装wsl
  3. 在Microsoft Store中找到ubuntu22LTS,并安装

note: 听说ubuntu24lts可能会遇到cuda兼容性问题

更改apt软件源(optional)

这个比较简单,没啥好说的

https://mirrors.tuna.tsinghua.edu.cn/help/ubuntu/

更改pip源(optional)

似乎ubuntu没有内置pip,需要手动安装。

1
sudo apt-get install python3-pip

然后更改pip源

1
pip config set global.index-url https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple

安装cuda

我选择的是cuda 12.8,下载wsl-ubuntu 专用版本

https://developer.nvidia.com/cuda-12-8-0-download-archive?target_os=Linux&target_arch=x86_64&Distribution=WSL-Ubuntu&target_version=2.0&target_type=deb_network

按照说明安装即可(网络版安装要简单点)

note:安装完成后,不知道为什么命令行中没有nvcc,千万不要去自己安装nvcc,因为nvcc其实是有的,手动安装反而容易造成不兼容。

安装jittor

  1. 首先创建并启用venv虚拟环境(可以使用vscode完成)
  2. 然后依次执行以下指令
pip install jittor
python -m jittor.test.test_example
# 如果您电脑包含Nvidia显卡,检查cudnn加速库
python -m jittor.test.test_cudnn_op

参考: https://cg.cs.tsinghua.edu.cn/jittor/download/

如果都显示OK,就安装成功了。