python代码小技巧

原创于 2021-03-30 11:48:26 发布 · 131 阅读

CC 4.0 BY-SA版权

9 篇文章

订阅专栏

部署运行你感兴趣的模型镜像

1. 指定GPU运行

import os 
os.environ["CUDA_VISIBLE_DEVICES"]="0,1"#指定0，1号GPU上运行代码

2. 查看GPU占用率

watch -n 1 -d nvidia-smi #每间隔1s刷新显示

3. 打印设计网络参数量

print ("Total number of paramerters in networks is {}  ".format (sum (x.numel () for x in model.parameters ())))

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch

Cuda

PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理