[問題]在Docker執行 使用GPU運算碰到的疑問

[Sam-good777]

大家好
這邊想問一個比較菜的問題

我是在Docker執行
目前想使用GPU進行運算
在作者的dockerhub裡面提到要--gpus=all(如下圖)

這邊我就不懂要怎麼執行了

顯卡是RTX4070 已經有安裝Nvidia驅動跟cuda toolkit
下圖是我目前用CPU運算的docker compose寫法
請問我該怎麼辦?

[frederik-uni]

I guess broke the docker files. The prompt is still in the old format. The prompt needs to be updated to the new format I guess