Jittor 环境配置
推荐环境:
Windows10 +Anaconda+VSCode
Anaconda配置
下载地址(内存少的可以安装miniconda)
1、配置环境变量
方法一
Anaconda安装时勾选 Add Anaconda to the system PATH environment variable
方法二 (手动添加)
验证:在Windows powershell中输入conda,提示以下则环境变量配置成功。
2、安装jittor
配置清华源:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
创建一个jittor的虚拟环境
conda create -n jittor python=3.8
安装jittor
python --version
conda install pywin32
python -m pip install jittor
测试是否安装成功
python -m jittor.test.test_core
python -m jittor.test.test_example
python -m jittor.test.test_cudnn_op
配置Vscode开发环境
1、插件安装
python
code runner
2、配置jittor环境
(1)快捷键ctrl+shift+p,搜索 Python:Select Interpreter
(2)选择刚刚配置好的jittor虚拟环境
3、环境测试
创建test.py
import jittor as jt
a = jt.float32([1,2,3])
print (a)
运行成功: