This article covers PyTorch's advanced GPU management features, how to optimise memory usage ... If you just call cuda , then the tensor is placed on GPU 0.
確定! 回上一頁