File tree Expand file tree Collapse file tree 2 files changed +2
-26
lines changed
unittest/windows_optdepts/scripts Expand file tree Collapse file tree 2 files changed +2
-26
lines changed Original file line number Diff line number Diff line change @@ -1305,35 +1305,11 @@ workflows:
1305
1305
name : unittest_linux_examples_gpu_py3.9
1306
1306
python_version : ' 3.9'
1307
1307
1308
- - unittest_windows_optdepts_cpu :
1309
- cu_version : cpu
1310
- name : unittest_windows_optdepts_cpu_py3.7
1311
- python_version : ' 3.7'
1312
1308
- unittest_windows_optdepts_cpu :
1313
1309
cu_version : cpu
1314
1310
name : unittest_windows_optdepts_cpu_py3.8
1315
1311
python_version : ' 3.8'
1316
- - unittest_windows_optdepts_cpu :
1317
- cu_version : cpu
1318
- name : unittest_windows_optdepts_cpu_py3.9
1319
- python_version : ' 3.9'
1320
- - unittest_windows_optdepts_cpu :
1321
- cu_version : cpu
1322
- name : unittest_windows_optdepts_cpu_py3.10
1323
- python_version : ' 3.10'
1324
- - unittest_windows_optdepts_gpu :
1325
- cu_version : cu116
1326
- name : unittest_windows_optdepts_gpu_py3.7
1327
- python_version : ' 3.7'
1328
1312
- unittest_windows_optdepts_gpu :
1329
1313
cu_version : cu116
1330
1314
name : unittest_windows_optdepts_gpu_py3.8
1331
1315
python_version : ' 3.8'
1332
- - unittest_windows_optdepts_gpu :
1333
- cu_version : cu116
1334
- name : unittest_windows_optdepts_gpu_py3.9
1335
- python_version : ' 3.9'
1336
- - unittest_windows_optdepts_gpu :
1337
- cu_version : cu116
1338
- name : unittest_windows_optdepts_gpu_py3.10
1339
- python_version : ' 3.10'
Original file line number Diff line number Diff line change 38
38
git submodule sync && git submodule update --init --recursive
39
39
40
40
printf " Installing PyTorch with %s\n" " ${cudatoolkit} "
41
- conda install -y -c " pytorch-${UPLOAD_CHANNEL} " -c nvidia " pytorch- ${UPLOAD_CHANNEL} " ::pytorch[build = " * ${version} * " ] " ${cudatoolkit} "
41
+ conda install -y -c " pytorch-${UPLOAD_CHANNEL} " -c nvidia pytorch " ${cudatoolkit} "
42
42
43
43
torch_cuda=$( python -c " import torch; print(torch.cuda.is_available())" )
44
44
echo torch.cuda.is_available is $torch_cuda
@@ -50,7 +50,7 @@ if [ ! -z "${CUDA_VERSION:-}" ] ; then
50
50
fi
51
51
fi
52
52
53
- pip install pip --upgrade
53
+ # python -m pip install pip --upgrade
54
54
55
55
# install tensordict
56
56
pip3 install git+https://github.com/pytorch-labs/tensordict
You can’t perform that action at this time.
0 commit comments