ما خوشحالیم که اعلام کنیم با شروع نسخه PyTorch 1.12، پشتیبانی AMD ROCm™ از بتا به Stable منتقل شده است. انتشار پایدار منجر به نگهداری طولانی مدت PyTorch در پشته نرم افزار ROCm، با ویژگی های عملکردی می شود که از پذیرش کاربر نهایی با ارزش افزوده اثبات شده پشتیبانی می کند. این یک گام تکاملی مهم برای AMD ROCm است زیرا پردازندههای گرافیکی AMD Instinct™ با یک بسته Python قابل نصب که در PyTorch.org میزبانی میشود، پشتیبانی جریان اصلی را به دست میآورند. کارتهای گرافیک AMD Radeon™ Pro W6800 برای توسعهدهندگان نیز پشتیبانی میشوند.
PyTorch یکی از محبوب ترین چارچوب های یادگیری ماشین منبع باز است که توسط محققان، دانشمندان، دانشگاهیان و طیف گسترده ای از صنایع استفاده می شود. همکاری قوی بین PyTorch و AMD برای ایجاد این تغییر به سمت پشتیبانی ROCm Stable وجود داشته است. “ما هیجان زده هستیم که با AMD همکاری کنیم تا پشتیبانی PyTorch خود را در ROCm افزایش دهیم و جامعه پر جنب و جوش PyTorch را قادر می سازد تا به سرعت آخرین نسل پردازنده های گرافیکی AMD Instinct را با عملکرد عالی برای موارد عمده استفاده از هوش مصنوعی که در PyTorch اجرا می شوند، بپذیرند.” سومیث چینتالا (سرپرست مهندس نرم افزار، PyTorch) می گوید
در حال نصب PyTorch
PyTorch را با استفاده از بسته Wheels نصب کنید
PyTorch با ارائه بسته های چرخ تست شده از پلت فرم ROCm پشتیبانی می کند. برای دسترسی به این بسته ها به ادامه مطلب مراجعه نمایید https://pytorch.org/get-started/locally/، و پلت فرم محاسبه ROCm را انتخاب کنید.
شکل 1: ماتریس نصب از Pytorch.org
دستورالعمل نصب زیر را دنبال کنید. نمونه ای از مراحل در زیر برای راحتی شما ارائه شده است.
1. یک تصویر پایه داکر با نسخه مناسب فضای کاربری ROCm نصب شده از https://hub.docker.com/r/rocm/dev-ubuntu-20.04 دریافت کنید یا یک تصویر داکر سیستم عامل پایه را دانلود کنید و ROCm را طبق دستورالعمل های نصب نصب کنید. . در این مثال، ROCm 5.1.1 نصب شده است که توسط ماتریس نصب از وب سایت pytorch.org پشتیبانی می شود.
docker pull rocm/dev-ubuntu-20.04:5.1.1
2. ظرف Docker را راه اندازی کنید.
docker run -it –device=/dev/kfd –device=/dev/dri –group-add video rocm/dev-ubuntu-20.04:5.1.1
3. هر گونه وابستگی مورد نیاز برای نصب چرخ ها را در داخل کانتینر داکر نصب کنید.
آپدیت مناسب
apt نصب libjpeg-dev python3-dev
pip3 نصب تنظیمات چرخ
4. مشعل، مشعل و مشعل را همانطور که در ماتریس نصب مشخص شده است نصب کنید.
pip3 نصب مشعل مشعل مشعل torchaudio –extra-index-url
https://download.pytorch.org/whl/rocm5.1.1
سایر گزینه ها و دستورالعمل ها برای نصب PyTorch از جمله داکر AMD Infinity Hub ظروف موجود در چارچوب های یادگیری عمیق (amd.com)
استفاده از Docker به شما قابلیت حمل و دسترسی به a کانتینر Docker از پیش ساخته شده است که به شدت در AMD آزمایش شده است. این همچنین ممکن است در زمان کامپایل صرفه جویی کند و باید دقیقاً همانطور که هنگام آزمایش انجام می شود عمل کند و از مشکلات احتمالی نصب جلوگیری کند.
نتیجه
پشتیبانی از پلتفرم نرم افزاری AMD ROCm همچنان در حال رشد و جلب توجه در صنعت است. با انتقال پشتیبانی از ROCm از بتا به پایدار، محققان و کاربران PyTorch میتوانند با استفاده از پردازندههای گرافیکی AMD Instinct و پشته نرمافزار ROCm با تمام ویژگیها و عملکردهای اصلی، به نوآوری خود ادامه دهند.
اطلاعات بیشتر
صفحه وب PyTorch: PyTorch
صفحه وب ROCm: AMD ROCm™ Open Software Platform | AMD
پورتال اطلاعات ROCm: AMD Documentation – Portal
AMD Instinct Accelerators: AMD Instinct™ Accelerators | AMD
AMD Infinity Hub: AMD Infinity Hub | AMD
ماهش بالاسوبرامانیان دیر است. Mgr بازاریابی محصول در واحد تجاری AMD Data Center GPU. پست های او نظرات خودش است و ممکن است موضع، استراتژی یا نظرات AMD را نشان ندهد. پیوندها به سایتهای شخص ثالث برای سهولت ارائه میشوند و AMD هیچ مسئولیتی در قبال محتویات این سایتهای لینکشده ندارد و هیچگونه تاییدیهای ندارد.