服务器上GPU的相关操作

1.用JupyterNotebook跑程序时,当发现某个包没有导入时,比如发现tqdm没有导入时,用下面的步骤导入:

(1)点击Running:

(2)点击terminals2,进入下面的页面:进行pip install tqdm即可

2.查看服务器上GPU的占用情况:

在上面展示的页面上输入:nvidia-smi

结果如上图所示

3.程序中指定使用的GPU节点区域:

os.environ["CUDA_VISIBLE_DEVICES"] = "5"#指定节点上面跑程序的区域

猜你喜欢

转载自blog.csdn.net/skj1995/article/details/82814387
今日推荐