如何查看jupyter notebook 运行占用cpu还是gpu
时间: 2024-09-20 18:15:30 浏览: 61
在Jupyter Notebook中,你可以通过运行特定的Python命令来检查计算资源的使用情况,包括CPU和GPU。对于CPU的使用情况,可以使用`%load_ext` 和 `%memit` 或者第三方库如`psutil`来查看。以下是查看CPU占用的一个简单示例:
```python
# 导入psutil库(如果未安装需先pip install psutil)
import psutil
# 获取当前进程的CPU使用率
cpu_usage = psutil.cpu_percent(interval=1)
print(f"当前CPU使用率: {cpu_usage}%")
# 对于GPU,如果你使用的是支持GPU的库,例如NumPy、Pandas或TensorFlow,并且已经在笔记本中激活了GPU计算,可以直接查看其使用情况。例如,在TensorFlow中:
import tensorflow as tf
gpus = tf.config.list_physical_devices('GPU')
if gpus:
for gpu in gpus:
print(f"GPU {gpu.name}:", tf.test.gpu_device_name())
else:
print("No GPU found.")
```