Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
torch_gpu.yml 172 Bytes
Newer Older
VoVAllen's avatar
[CI] Change tests for flexibility  
VoVAllen committed
1
2 
name: pytorch-ci
dependencies:
VoVAllen's avatar
[Backend] TF backend (#978)  
VoVAllen committed
3 
  - python=3.6.9
VoVAllen's avatar
[CI] Change tests for flexibility  
VoVAllen committed
4 
  - pip
VoVAllen's avatar
[Backend] TF backend (#978)  
VoVAllen committed
5 
  - pip:
Jinjing Zhou's avatar
[CI] Update PyTorch in CI image to 1.5.1 (#1808)  
Jinjing Zhou committed
6
7 
    - --find-links https://download.pytorch.org/whl/torch_stable.html
    - --requirement torch_gpu_pip.txt