File tree Expand file tree Collapse file tree 2 files changed +6
-4
lines changed Expand file tree Collapse file tree 2 files changed +6
-4
lines changed Original file line number Diff line number Diff line change @@ -24,10 +24,10 @@ pip install --progress-bar off -r $DIR/../requirements.txt
24
24
25
25
# Install PyTorch Nightly for test.
26
26
# Nightly - pip install --pre torch torchvision torchaudio -f https://download.pytorch.org/whl/nightly/cu102/torch_nightly.html
27
- # Install 2.1 for testing
28
- # pip uninstall -y torch torchvision torchaudio torchtext torchdata
29
- # pip3 install torch torchvision torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu121
30
- # pip3 install torchdata torchtext --index-url https://download.pytorch.org/whl/test/cpu
27
+ # Install 2.2 for testing
28
+ pip uninstall -y torch torchvision torchaudio torchtext torchdata
29
+ pip3 install torch==2.2.0 torchvision torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu121
30
+ pip3 install torchdata torchtext --index-url https://download.pytorch.org/whl/test/cpu
31
31
32
32
# Install two language tokenizers for Translation with TorchText tutorial
33
33
python -m spacy download en_core_web_sm
Original file line number Diff line number Diff line change 27
27
"intermediate_source/mnist_train_nas" , # used by ax_multiobjective_nas_tutorial.py
28
28
"intermediate_source/fx_conv_bn_fuser" ,
29
29
"intermediate_source/_torch_export_nightly_tutorial" , # does not work on release
30
+ "intermediate_source/torch_export_tutorial" , # Enable when fixed for 2.2
30
31
"advanced_source/super_resolution_with_onnxruntime" ,
31
32
"advanced_source/ddp_pipeline" , # requires 4 gpus
32
33
"prototype_source/fx_graph_mode_ptq_dynamic" ,
34
+ "prototype_source/maskedtensor_sparsity" , # Enable when fixed for 2.2
33
35
"prototype_source/vmap_recipe" ,
34
36
"prototype_source/torchscript_freezing" ,
35
37
"prototype_source/nestedtensor" ,
You can’t perform that action at this time.
0 commit comments