Dstack-TEE / Dstack-TEE/dstack
Add a tool to config NVIDIA PPCIE/CC mode manually and automatcially
- Ngôn ngữ chính
- Rust
- Star
- 544
- Fork
- 96
- Merge trung bình
- 23 giờ 40 phút
- Pull request đã merge (30 ngày)
- 126
Mô tả
Current we use those command to switch modes:
PPCIE mode:
```
for i in $(seq 0 $(($(lspci -nn | grep -c "10de") - 1))); do sudo python3 ./nvidia_gpu_tools.py --set-cc-mode=off --reset-after-cc-mode-switch --gpu=$i; done
for i in $(seq 0 $(($(lspci -nn | grep -c "10de") - 1))); do sudo python3 ./nvidia_gpu_tools.py --set-ppcie-mode=on --reset-after-ppcie-mode-switch --gpu=$i; done
```
CC mode:
```
for i in $(seq 0 $(($(lspci -nn | grep -c "10de") - 1))); do sudo python3 ./nvidia_gpu_tools.py --set-ppcie-mode=off --reset-after-ppcie-mode-switch --gpu=$i; done
for i in $(seq 0 $(($(lspci -nn | grep -c "10de") - 1))); do sudo python3 ./nvidia_gpu_tools.py --set-cc-mode=on --reset-after-cc-mode-switch --gpu=$i; done
```
Disable CC at all:
```
for i in $(seq 0 $(($(lspci -nn | grep -c "10de") - 1))); do sudo python3 ./nvidia_gpu_tools.py --set-cc-mode=off --reset-after-cc-mode-switch --gpu=$i; done
for i in $(seq 0 $(($(lspci -nn | grep -c "10de") - 1))); do sudo python3 ./nvidia_gpu_tools.py --set-ppcie-mode=off --reset-after-ppcie-mode-switch --gpu=$i; done
```
We should add it to dstack. So users don't need to manually install nvidia_gpu_tools.
Hướng dẫn đóng góp
Đánh giá
Issue này chưa được đánh giá.