cc_staff
284
edits
PyTorch: Difference between revisions
m
Fixed module load command
m (forgot to "make") |
m (Fixed module load command) |
||
| Line 2,141: | Line 2,141: | ||
<tabs> | <tabs> | ||
<tab name="StdEnv/2023"> | <tab name="StdEnv/2023"> | ||
module load StdEnv/2023 gcc cuda/12.2 cmake protobuf cudnn python/3.11 | module load StdEnv/2023 gcc cuda/12.2 cmake protobuf cudnn python/3.11 abseil cusparselt opencv/4.8.1 | ||
virtualenv --no-download --clear ~/ENV && source ~/ENV/bin/activate | virtualenv --no-download --clear ~/ENV && source ~/ENV/bin/activate | ||
pip install --no-index torch numpy | pip install --no-index torch numpy | ||
Note that the versions for the abseil, cusparselt and opencv modules may need to be adjusted, | Note that the versions for the abseil, cusparselt and opencv modules may need to be adjusted, | ||