diff --git a/conda/environments/cuda_dev.yaml b/conda/environments/cuda_dev.yaml
index e092385..f82d0f0 100644
--- a/conda/environments/cuda_dev.yaml
+++ b/conda/environments/cuda_dev.yaml
@@ -20,13 +20,15 @@ dependencies:
   - pytest
   - pytest-benchmark
   - pytest-cov>=2
-  - pytorch>=2.0
+  - pytorch>=2.0=*cuda*
+  - tiktoken
   - transformers
   - tokenizers
   - torchvision
   - scikit-learn>=0.23.1
   - scipy
   - setuptools>=61.0.0
+  - sentencepiece
   - sentence-transformers
   - pip:
     - pytrec-eval
diff --git a/requirements/cuda12x.txt b/requirements/cuda12x.txt
index 54e1c2d..24632b9 100644
--- a/requirements/cuda12x.txt
+++ b/requirements/cuda12x.txt
@@ -11,3 +11,4 @@ curated-transformers>=1.0
 bitsandbytes>=0.30
 sentence-transformers>=2.0
 sentencepiece
+tiktoken