File tree 4 files changed +13
-16
lines changed
4 files changed +13
-16
lines changed Original file line number Diff line number Diff line change @@ -14,7 +14,7 @@ tqdm==4.66.1
14
14
numpy == 1.24.4
15
15
matplotlib
16
16
librosa
17
- torch == 2.6
17
+ torch == 2.7
18
18
torchvision
19
19
torchdata
20
20
networkx
67
67
pygame == 2.6.0
68
68
pycocotools
69
69
semilearn == 0.3.2
70
- torchao == 0.5 .0
70
+ torchao == 0.10 .0
71
71
segment_anything == 1.0
72
72
torchrec == 1.1.0 ; platform_system == "Linux"
73
- fbgemm-gpu == 1.1 .0 ; platform_system == "Linux"
73
+ fbgemm-gpu == 1.2 .0 ; platform_system == "Linux"
Original file line number Diff line number Diff line change @@ -22,13 +22,10 @@ sudo apt-get install -y pandoc
22
22
# Install PyTorch Nightly for test.
23
23
# Nightly - pip install --pre torch torchvision torchaudio -f https://download.pytorch.org/whl/nightly/cu102/torch_nightly.html
24
24
# Install 2.5 to merge all 2.4 PRs - uncomment to install nightly binaries (update the version as needed).
25
- # sudo pip uninstall -y torch torchvision torchaudio torchtext torchdata
26
- # sudo pip3 install torch==2.6.0 torchvision --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu124
27
25
# sudo pip uninstall -y fbgemm-gpu torchrec
26
+ # sudo pip uninstall -y torch torchvision torchaudio torchtext torchdata torchrl tensordict
28
27
# sudo pip3 install fbgemm-gpu==1.1.0 torchrec==1.0.0 --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu124
29
- sudo pip uninstall -y torch torchvision torchaudio torchtext torchdata torchrl tensordict
30
- pip3 install torch==2.7.0 torchvision torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu126
31
- # sudo pip uninstall -y fbgemm-gpu
28
+ # pip3 install torch==2.7.0 torchvision torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu126
32
29
# Install two language tokenizers for Translation with TorchText tutorial
33
30
python -m spacy download en_core_web_sm
34
31
python -m spacy download de_core_news_sm
Original file line number Diff line number Diff line change 51
51
"intermediate_source/text_to_speech_with_torchaudio" ,
52
52
"intermediate_source/tensorboard_profiler_tutorial" , # reenable after 2.0 release.
53
53
"advanced_source/semi_structured_sparse" , # reenable after 3303 is fixed.
54
- "intermediate_source/mario_rl_tutorial" , # reenable after 3302 is fixed
55
- "intermediate_source/reinforcement_ppo" , # reenable after 3302 is fixed
56
- "intermediate_source/pinmem_nonblock" , # reenable after 3302 is fixed
57
- "intermediate_source/dqn_with_rnn_tutorial" , # reenable after 3302 is fixed
58
- "advanced_source/pendulum" , # reenable after 3302 is fixed
59
- "advanced_source/coding_ddpg" , # reenable after 3302 is fixed
60
- "intermediate_source/torchrec_intro_tutorial" , # reenable after 3302 is fixed
61
- "recipes_source/recipes/reasoning_about_shapes" # reenable after 3326 is fixed
54
+ "intermediate_source/torchrec_intro_tutorial" , # reenable after 3302 is fixe
55
+ "intermediate_source/memory_format_tutorial" , # causes other tutorials like torch_logs fail. "state" issue, reseting dynamo didn't help
62
56
]
63
57
64
58
def tutorial_source_dirs () -> List [Path ]:
Original file line number Diff line number Diff line change 99
99
100
100
def reset_seeds (gallery_conf , fname ):
101
101
torch .cuda .empty_cache ()
102
+ torch .backends .cudnn .deterministic = True
103
+ torch .backends .cudnn .benchmark = False
104
+ torch ._dynamo .reset ()
105
+ torch ._inductor .config .force_disable_caches = True
102
106
torch .manual_seed (42 )
103
107
torch .set_default_device (None )
104
108
random .seed (10 )
105
109
numpy .random .seed (10 )
110
+ torch .set_grad_enabled (True )
111
+
106
112
gc .collect ()
107
113
108
114
sphinx_gallery_conf = {
You can’t perform that action at this time.
0 commit comments