Cách cài đặt CUDA trên Ubuntu để tăng hiệu năng train model

Hi anh em, Hi anh em,hôm nay chúng ta cùng tìm hiểu cách cài đặt CUDA trên Ubuntu để tăng hiệu năng cho máy chủ khi train model nhé!

cuda trên ubuntu
Nguồn: Tại đây

Nhân dịp ra mắt clip review về dịch vụ cho thuê máy chủ GPU của iRender.vn nên mình ra bài này luôn. Bài này mình có test thì phù hợp cả cho Ubuntu 18.04 và 20 nên các bạn cứ thử xem nhé!

Phần 1 – Xóa các bản cài đặt cũ

Các bạn mở Terminal trong Ubuntu và gõ lệnh sau để gỡ bỏ các bản cài cũ nếu có:

Phần 2 – Cài đặt source PPA cho CUDA

Các bạn cứ gõ lần lượt từng lệnh một cho tới hết, nếu có lỗi thì dừng lại và tìm phương án xử lý hoặc post lên https://www.facebook.com/groups/miaigroup để trao đổi, thảo luận nhé!

Phần 3 – Tiến hành cài đặt CUDA trên Ubuntu

Cài đặt các gói CUDA và CuDNN:

Sau đó bạn thêm PATH cho CUDA vào ./profile:

Và thêm đoạn này vào cuối file:

Sau khi chạy lệnh này bạn restart máy chủ Ubuntu!

Nguồn: Tại đây

Phần 4 – Kiểm tra bộ cài đặt CUDA trên Ubuntu

Bạn gõ lệnh kiểm tra phiên bản CUDA:

Kiểm tra NVIDA Driver và card màn hình:

Nếu thấy hiện lên cửa sổ “giống giống” như bên dưới là okie:

cuda trên ubuntu
Nguồn: Tại đây

Sau bước này thì các bạn đã cài đặt CUDA trên Ubuntu có thể train model trên GPU ầm ầm rồi.

Nếu như còn thiếu pip, git để cài các package cho Python (như opencv-python, torch…) thì bạn cài tiếp:

Chúc các bạn thành công! Nếu có lỗi thì dừng lại và tìm phương án xử lý hoặc post lên https://www.facebook.com/groups/miaigroup để trao đổi, thảo luận nhé!

#MìAI

Fanpage: http://facebook.com/miaiblog
Group trao đổi, chia sẻ: https://www.facebook.com/groups/miaigroup
Website: https://miai.vn
Youtube: http://bit.ly/miaiyoutube

Tham khảo: https://medium.com/@exesse/cuda-10-1-installation-on-ubuntu-18-04-lts-d04f89287130

Related Post

Leave a Reply

Your email address will not be published. Required fields are marked *