創(chuàng)新互聯(lián)www.cdcxhl.cn八線動(dòng)態(tài)BGP香港云服務(wù)器提供商，新人活動(dòng)買多久送多久，劃算不套路！

婺城ssl適用于網(wǎng)站、小程序/APP、API接口等需要進(jìn)行數(shù)據(jù)傳輸應(yīng)用場景，ssl證書未來市場廣闊！成為成都創(chuàng)新互聯(lián)的ssl證書銷售渠道，可以享受市場價(jià)格4-6折優(yōu)惠！如果有意向歡迎電話聯(lián)系或者加微信：13518219792（備注：SSL證書合作）期待與您的合作！

小編給大家分享一下如何設(shè)置多卡服務(wù)器下隱藏部分GPU和TensorFlow的顯存使用，希望大家閱讀完這篇文章后大所收獲，下面讓我們一起去探討方法吧！

服務(wù)器有多張顯卡，一般是組里共用，分配好顯卡和任務(wù)就體現(xiàn)公德了。除了在代碼中指定使用的 GPU 編號(hào)，還可以直接設(shè)置可見 GPU 編號(hào)，使程序/用戶只對(duì)部分 GPU 可見。

操作很簡單，使用環(huán)境變量 CUDA_VISIBLE_DEVICES 即可。

具體來說，如果使用單卡運(yùn)行 Python 腳本，則可輸入

CUDA_VISIBLE_DEVICES=1 python my_script.py

腳本將只使用 GPU1。

在 .py 腳本和 Notebook 中設(shè)置，則

import os

os.environ["CUDA_VISIBLE_DEVICES"] = "2"

還可以直接設(shè)置臨時(shí)的環(huán)境變量：

export CUDA_VISIBLE_DEVICES="0"

此時(shí)該用戶的 CUDA 只看得見 GPU0。

至于顯存設(shè)置，可以設(shè)置使用比例(70%)：

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7)

sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

也可以按需增長：

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config, ...)

網(wǎng)站標(biāo)題：如何設(shè)置多卡服務(wù)器下隱藏部分GPU和TensorFlow的顯存使用-創(chuàng)新互聯(lián)
分享地址：http://aaarwkj.com/article28/dophjp.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián)，為您提供域名注冊(cè)、網(wǎng)站建設(shè)、網(wǎng)站營銷、服務(wù)器托管、建站公司、虛擬主機(jī)

廣告

聲明：本網(wǎng)站發(fā)布的內(nèi)容（圖片、視頻和文字）以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主，如果涉及侵權(quán)請(qǐng)盡快告知，我們將會(huì)在第一時(shí)間刪除。文章觀點(diǎn)不代表本網(wǎng)站立場，如需處理請(qǐng)聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載，或轉(zhuǎn)載時(shí)需注明來源： 創(chuàng)新互聯(lián)