Ôõô¿´linuxµÄpytorchÇéÐÎ
ÒªÉó²é linux É쵀 pytorch ÇéÐΣ¬¿ÉÒÔÒÀ´ÎÖ´ÐÐÒÔÏ°취£º¼ì²é pytorch °æ±¾Éó²éÒÑ×°ÖÃµÄ pytorch °üºÍ°æ±¾¼ì²é cuda °æ±¾£¨ÈôÊÇÒÑ×°Öã©Éó²é cudnn °æ±¾£¨ÈôÊÇÒÑ×°Ö㩼ì²é openmp Ö§³Ö¼ì²é gpu ¿ÉÓÃÐÔÉó²éÇéÐαäÁ¿¼ì²é python °æ±¾¼ì²é pip °æ±¾¼ì²éϵͳÐÅÏ¢
ÔõÑùÉó²é Linux É쵀 PyTorch ÇéÐΣ¿
ÒªÔÚ Linux ÉÏÉó²é PyTorch ÇéÐΣ¬Çëƾ֤ÒÔÏ°취²Ù×÷£º
1. ¼ì²é PyTorch °æ±¾£¨¿ÉÑ¡£©£º
python -c "import torch; print(torch.__version__)"
µÇ¼ºó¸´ÖÆ
2. Éó²éÒÑ×°ÖÃµÄ PyTorch °üºÍ°æ±¾£º
pip list | grep torch
µÇ¼ºó¸´ÖÆ
3. ¼ì²é CUDA °æ±¾£¨ÈôÊÇÒÑ×°Öã©£º
nvcc --version
µÇ¼ºó¸´ÖÆ
4. Éó²é cuDNN °æ±¾£¨ÈôÊÇÒÑ×°Öã©£º
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR
µÇ¼ºó¸´ÖÆ
5. ¼ì²é OpenMP Ö§³Ö£º
python -c "import torch; print(torch.backends.openmp.is_available())"
µÇ¼ºó¸´ÖÆ
6. ¼ì²é GPU ¿ÉÓÃÐÔ£º
nvidia-smi
µÇ¼ºó¸´ÖÆ
7. Éó²éÇéÐαäÁ¿£º
env | grep CUDA env | grep CUDNN
µÇ¼ºó¸´ÖÆ
8. ¼ì²é Python °æ±¾£º
python --version
µÇ¼ºó¸´ÖÆ
9. ¼ì²é pip °æ±¾£º
pip --version
µÇ¼ºó¸´ÖÆ
10. ¼ì²éϵͳÐÅÏ¢£º
uname -a cat /etc/os-release
µÇ¼ºó¸´ÖÆ
ÒÔÉϾÍÊÇÔõô¿´linuxµÄpytorchÇéÐεÄÏêϸÄÚÈÝ£¬¸ü¶àÇë¹Ø×¢±¾ÍøÄÚÆäËüÏà¹ØÎÄÕ£¡