python - টেনসরফ্লো বৃহস্পতির মধ্যে CUDA_VISIBLE_DEVICES সেট করে




environment-variables tensorflow (2)

আপনি কেবল magics সাহায্যে কোনও আমদানি ছাড়াই এটি দ্রুত করতে পারেন:

%env CUDA_DEVICE_ORDER=PCI_BUS_ID
%env CUDA_VISIBLE_DEVICES=0

লক্ষ্য করুন যে সমস্ত এনভের্নি ভেরিয়েবল স্ট্রিং, সুতরাং ব্যবহারের দরকার নেই " You env-ভেরিয়েবলটি চালিয়ে যাচাই করতে পারবেন: %env <name_of_var> বা সমস্ত %env সাথে চেক করুন।

আমার দুটি জিপিইউ রয়েছে এবং আইপিনবের মাধ্যমে একই সাথে দুটি আলাদা নেটওয়ার্ক চালনা করতে চাই, তবে প্রথম নোটবুকটি সর্বদা উভয় জিপিইউ বরাদ্দ করে।

CUDA_VISIBLE_DEVICES ব্যবহার করে আমি পাইথন ফাইলগুলির জন্য ডিভাইসগুলি আড়াল করতে পারি, তবে আমি কীভাবে নোটবুকের মধ্যে এটি করব তা সম্পর্কে আমি নিশ্চিত নই।

একই সার্ভারে চলছে নোটবুকগুলিতে বিভিন্ন জিপিইউ লুকিয়ে রাখার উপায় আছে কি?


আপনি নোটবুকে os.environ ব্যবহার করে পরিবেশের পরিবর্তনগুলি সেট করতে পারেন os.environ টেনসরফ্লো প্রথম জিপিইউতে সীমাবদ্ধ করতে টেনসরফ্লো শুরু করার আগে নিম্নলিখিতটি করুন।

import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"   # see issue #152
os.environ["CUDA_VISIBLE_DEVICES"]="0"

আপনি ডাবল পরীক্ষা করতে পারেন যে আপনার কাছে সঠিক ডিভাইস টিএফ-এর কাছে দৃশ্যমান

from tensorflow.python.client import device_lib
print device_lib.list_local_devices()

আমি নোটবুক_উটিলের মতো ইউটিলিটি মডিউল থেকে এটি ব্যবহার করার ঝোঁক

import notebook_util
notebook_util.pick_gpu_lowest_memory()
import tensorflow as tf




tensorflow