简介

刚入门的时候大部分人可能还是会选择用windows来装deep learning的框架，刚好换了电脑，这里从零开始演示一下Jittor框架在win10下的安装流程。

Jittor框架安装

先上jittor官网:https://cg.cs.tsinghua.edu.cn/jittor/

右上方进入安装，选择windows：

如果只有CPU的话Platform选CPU就可以，我这儿还没装python，先到python官网装对应版本的python，这里版本显示大于等于3.8即可。

目前有windows installer的最新版是3.10.1，python版本只要在3.8以上，对本次安装和后面的样例不会有任何影响，我们这里选3.9.7版本的python。选“Windows installer (64-bit)”版本，比较好装。

下载之后双击安装，Add to PATH勾一下，最后Disable path length limit 也点一下避免以后一些不必要的麻烦，之后就好了：

然后继续按官方教程装jittor：

python -m pip install jittor
python -m jittor.test.test_core
python -m jittor.test.test_example
python -m jittor.test.test_cudnn_op

如果装的太慢可以换清华源，比如：

python -m pip install jittor -i https://pypi.tuna.tsinghua.edu.cn/simple

这里版本我们选择1.3.1.18。

python -m pip install jittor==1.3.1.18 -i https://pypi.tuna.tsinghua.edu.cn/simple

关于版本问题这里简单说一下，因为windows系统的原因，每次jittor的小更新可能都会造成在windows下无法正常安装（一般jittor.test.test_core那边会出问题），比如之前issue里就有一个：https://github.com/Jittor/jittor/issues/278 ，这样的问题定位挺浪费时间的，有条件还是熟悉下Linux，毕竟windows终究只是入门玩玩。1.3.1.18这个版本windows下测下来比较稳定。

后面3条命令依次跑完就可以了。

到这里cpu版的就测试完了，有cuda的最后跑一下cudnn_op测试就可以了：

最后可以看到2个test fail了，看一下array里面的值，最终误差很小（估摸着小于0.01%），不用管，而且都是3d相关的计算。到这里Windows下的Jittor就装完了，大部分人一开始可能碰到的Jittor在windows下安装失败可能都是因为小版本更新带来的问题，比如直接pip install之后第一次test就挂了。如果是cuda的问题的话要检查一下cuda版本是否在10.2以上，如果是另装cuda的话这里推荐直接11.3版本及以上的。

其他

因为很多预训练模型会选择用pytorch里面的（比如swin transformer，efficient-net系列等），要迁移这些模型的话还是得再装一下pytorch，或者自己下了模型按官方教程读取。Windows下如果直接用官方文档提供的事例代码迁移可能会报一些错，ubuntu版本我自己试了一下都没什么问题。推荐直接用jittor原生实现的模型即可，比如：

import jittor as jt
from jittor.models import resnet50
jittor_model = resnet50()
dict = jittor_model.state_dict()
jittor_model.load_state_dict(dict)

Jittor（计图）深度学习框架安装相关推荐

2020CADCG专题报告笔记 Jittor计图深度学习框架
CV：Win10下深度学习框架安装之Tensorflow/tensorflow_gpu+Cuda+Cudnn(最清楚/最快捷)之详细攻略(图文教程)
CV:Win10下深度学习框架安装之Tensorflow/tensorflow_gpu+Cuda+Cudnn(最清楚/最快捷)之详细攻略(图文教程) 导读本人在Win10下安装深度学习框架Tenso ...
盘点国内那些深度学习框架：清华计图Jittor、腾讯优图NCNN、百度飞桨PaddlePaddle、阿里X-DeepLearning
1.清华计图Jittor 清华大学开发了一个名为计图(Jittor)的深度学习框架. 计图(Jittor:Just in Time)是一个采用元算子表达神经网络计算单元.完全基于动态编译(Just-i ...
清华开源 Jittor：首个国内高校自研深度学习框架，一键转换 PyTorch
点击上方"视学算法",马上关注真爱,请设置"星标"或点个"在看" 来源 | 机器之心继 Theano.Caffe 之后,又一个由高校主导 ...
清华开源Jittor：首个国内高校自研深度学习框架，一键转换PyTorch
关注上方"深度学习技术前沿",选择"星标公众号", 资源干货,第一时间送达! 来源:机器之心@微信公众号继 Theano.Caffe 之后,又一个由高校主导的 ...
Euler 今日问世！国内首个工业级的图深度学习开源框架，阿里妈妈造
千呼万唤始出来!阿里妈妈正式公布重磅开源项目--图深度学习框架Euler.这是国内首个在核心业务大规模应用后开源的图深度学习框架.此次开源,Euler内置了大量的算法供用户直接使用,相关代码已经可在G ...
清华「计图」、旷视「天元」纷纷重磅开源，国产深度学习框架迎来高光时刻...
来源:CSDN 本文约3141字,建议阅读7分钟. 本文介绍清华开源的深度学习框架 Jittor(计图) 引起了广泛关注,支撑旷视诸多业务和战绩的深度学习框架 MegEngine(天元) 也将在近日开 ...
清华开源深度学习框架计图，开源超级玩家再进阶
2020-03-22 09:24 导语:清华开源计图,背后是三代人的共同努力. 雷锋网AI源创评论报道,据官方消息,清华大学计算机系图形实验室宣布开源一个全新的深度学习框架:Jittor,中文名计图. ...
清华自研深度学习框架「计图」开源！多项任务性能超过PyTorch
2020-03-20 11:42:07 乾明发自凹非寺量子位报道 | 公众号 QbitAI AI框架,又来重磅中国玩家. 刚刚,清华自研的深度学习框架,正式对外开源."贵系&quo ...

Jittor（计图）深度学习框架安装

Jittor（计图）深度学习框架安装

简介

Jittor框架安装

其他

Jittor（计图）深度学习框架安装相关推荐

最新文章

热门文章