在WSL中安装计图Jittor
众所周知,jittor对windows的支持是一言难尽,各种cuda问题层出不穷。
为了在windows上使用jittor,需要在wsl中安装jittor。
安装 WSL : ubuntu22LTS
- 启用hyper-v
- 安装wsl
- 在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 专用版本
按照说明安装即可(网络版安装要简单点)
note:安装完成后,不知道为什么命令行中没有nvcc,千万不要去自己安装nvcc,因为nvcc其实是有的,手动安装反而容易造成不兼容。
安装jittor
- 首先创建并启用venv虚拟环境(可以使用vscode完成)
- 然后依次执行以下指令
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,就安装成功了。
本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议。转载请注明来源 57U's Blog!
评论