File tree Expand file tree Collapse file tree 1 file changed +4
-8
lines changed Expand file tree Collapse file tree 1 file changed +4
-8
lines changed Original file line number Diff line number Diff line change @@ -10,22 +10,18 @@ export PYTORCH_CUDA_PKG=""
10
10
export CONDA_ENV=" build_binary"
11
11
12
12
if [[ ${MATRIX_PYTHON_VERSION} = ' 3.13t' ]]; then
13
- echo " Conda doesn't support 3.13t yet, you can just try \` conda create -n test python=3.13t\` "
14
- exit 0
13
+ # use conda-forge to install python3.13t
14
+ conda create -y -n " ${CONDA_ENV} " python=" 3.13" python-freethreading -c conda-forge
15
+ else
16
+ conda create -y -n " ${CONDA_ENV} " python=" ${MATRIX_PYTHON_VERSION} "
15
17
fi
16
18
17
- conda create -y -n " ${CONDA_ENV} " python=" ${MATRIX_PYTHON_VERSION} "
18
-
19
19
conda run -n build_binary python --version
20
20
21
21
# Install pytorch, torchrec and fbgemm as per
22
22
# installation instructions on following page
23
23
# https://github.com/pytorch/torchrec#installations
24
24
25
- if [[ ${MATRIX_GPU_ARCH_TYPE} = ' rocm' ]]; then
26
- echo " We don't support rocm"
27
- exit 0
28
- fi
29
25
30
26
# figure out CUDA VERSION
31
27
if [[ ${MATRIX_GPU_ARCH_TYPE} = ' cuda' ]]; then
You can’t perform that action at this time.
0 commit comments