python實現自動查詢並使用空閒GPU

2021-08-07 21:26:17 字數 580 閱讀 6455

檢視gpu memory,並將結果儲存在tmp中:

os.system('nvidia-smi -q -d memory |grep -a4 gpu|grep free >tmp')
讀取gpu memory:

求剩餘memory最多的顯示卡號,並設定cuda_visible_devices為該顯示卡:

os.environ['cuda_visible_devices']=str(np.argmax(memory_gpu))

os.system('rm tmp')

這裡顯示卡1剩餘memory最多,因此cuda_visible_devices=1,即使用1號顯示卡.

執行**,終端輸入nvidia-smi,即可檢視顯示卡使用資訊.

Python3爬蟲 自動查詢天氣並實現語音播報

要獲取本機ip,這裡有乙個很簡單的辦法,就是開啟這個網頁 開啟之後就能看到我們的ip位址和所在地理位置資訊了。因為我們可以請求一下這個網頁,然後對返回的結果進行一下解析,就能得到我們想要的結果了。如下 1 2 version python3.5 3author onion 4site 5time 2...

Python3爬蟲之自動查詢天氣並實現語音播報

1 get ip.py 要獲取本機ip,這裡有乙個很簡單的辦法,就是開啟這個網頁 開啟之後就能看到我們的ip位址和所在地理位置資訊了。因為我們可以請求一下這個網頁,然後對返回的結果進行一下解析,就能得到我們想要的結果了。如下 import re import requests 獲取本機ip和地理位置...

Python3爬蟲之自動查詢天氣並實現語音播報

一 寫在前面 之前寫過一篇用python傳送天氣預報郵件的部落格,但是因為要手動輸入城市名稱,還要開啟郵箱才能知道天氣情況,這也太麻煩了。於是乎,有了這一篇部落格,這次我要做的就是用python獲取本機ip位址,並根據這個ip位址獲取物理位置也就是我所在的城市名稱,然後用之前的辦法實現查詢天氣,再利...