HomeBlogCác loại khácCách cài đặt PyTorch trên Windows 10 / 11 [Nvidia AMD GPU & CPU]

Cách cài đặt PyTorch trên Windows 10 / 11 [Nvidia AMD GPU & CPU]

  • avatarJoao Silva
  • 2025-04-28 19:15
  • 3 Đọc trong giây phút
cover_img
  1. Các yêu cầu trước khi cài đặt PyTorch
  2. Điều hướng trang cài đặt PyTorch
  3. Chọn nền tảng tính toán phù hợp
  4. Kiểm tra phiên bản CUDA
  5. Tải xuống và cài đặt CUDA
  6. Hoàn tất cài đặt CUDA
  7. Cài đặt Visual Studio
  8. Hoàn tất cài đặt CUDA
  9. Cài đặt PyTorch
  10. Xác minh cài đặt PyTorch
  11. Câu hỏi thường gặp

Các yêu cầu trước khi cài đặt PyTorch

Trước khi cài đặt PyTorch, hãy đảm bảo rằng Python và pip đã được cài đặt trên máy của bạn. Nếu bạn chưa cài đặt chúng, hãy tham khảo các nguồn tài nguyên trực tuyến đáng tin cậy để biết hướng dẫn cách thực hiện. Khi bạn đã thiết lập Python và pip, bạn có thể tiến hành cài đặt PyTorch.

Điều hướng trang cài đặt PyTorch

Để bắt đầu quá trình cài đặt, hãy truy cập trang web chính thức của PyTorch. Cuộn xuống để tìm các tùy chọn cài đặt. Chọn phiên bản ổn định của PyTorch 2.1.1, và chọn hệ điều hành của bạn. Đối với người dùng Windows, chọn 'Windows' làm hệ điều hành, và vì bạn không có Conda, hãy giữ tùy chọn gói là pip. Ngôn ngữ nên giữ nguyên là Python.

Chọn nền tảng tính toán phù hợp

Tùy chọn nền tảng tính toán liên quan đến loại card đồ họa của bạn. Nếu bạn có card đồ họa Nvidia, hãy chọn phiên bản CUDA phù hợp. Đối với card đồ họa AMD, chọn ROCm. Nếu hệ thống của bạn không có GPU, hãy chọn tùy chọn CPU. Lưu ý rằng tùy chọn ROCm không có sẵn cho Windows; nó chỉ có thể được truy cập trên Linux.

Kiểm tra phiên bản CUDA

Trước khi thực hiện lệnh pip install, hãy kiểm tra phiên bản CUDA đã cài đặt của bạn. Mở PowerShell và nhập lệnh 'nvcc --version'. Nếu CUDA chưa được cài đặt, bạn sẽ cần tải xuống từ trang web Nvidia. Tìm kiếm 'cài đặt CUDA 12.1' và nhấp vào kết quả đầu tiên để truy cập tải xuống bộ công cụ.

Tải xuống và cài đặt CUDA

Trên trang tải xuống CUDA, chọn hệ điều hành và kiến trúc của bạn (x86_64), và xác nhận phiên bản Windows của bạn. Chọn loại trình cài đặt 'exe local' và đảm bảo bạn có ít nhất 3.1 GB dung lượng trống trước khi bắt đầu tải xuống. Khi tải xuống hoàn tất, hãy điều hướng đến thư mục tải xuống của bạn và chạy tệp thực thi.

Hoàn tất cài đặt CUDA

Sau khi khởi động trình cài đặt CUDA, hãy làm theo các hướng dẫn để giải nén các tệp. Đồng ý với thỏa thuận cấp phép và tiếp tục với cài đặt nhanh. Nếu được hỏi về khả năng tương thích với Visual Studio, bạn sẽ cần cài đặt Visual Studio để đảm bảo đầy đủ chức năng.

Cài đặt Visual Studio

Để cài đặt Visual Studio, hãy tìm kiếm nó trực tuyến và điều hướng đến trang tải xuống. Tải xuống phiên bản Community, miễn phí. Khi tải xuống xong, mở trình cài đặt và chọn các gói bạn muốn cài đặt. Nên bao gồm gói phát triển Python. Sau khi chọn các gói của bạn, nhấp vào nút cài đặt.

Hoàn tất cài đặt CUDA

Khi Visual Studio đã được cài đặt, quay lại trình cài đặt CUDA và chạy tệp thực thi một lần nữa. Làm theo các bước giải nén và cài đặt giống như trước. Sau khi cài đặt hoàn tất, xác minh phiên bản CUDA bằng cách chạy 'nvcc --version' trong PowerShell. Bạn sẽ thấy rằng CUDA 12.1 đã được cài đặt.

Cài đặt PyTorch

Với CUDA đã được cài đặt thành công, bạn có thể tiến hành cài đặt PyTorch. Mở PowerShell và chạy lệnh pip phù hợp để cài đặt PyTorch. Điều này sẽ tải xuống và thiết lập PyTorch trên hệ điều hành Windows 10 hoặc 11 của bạn.

Xác minh cài đặt PyTorch

Để đảm bảo rằng PyTorch đã được cài đặt đúng cách, hãy chạy một vài lệnh trong PowerShell. Bắt đầu bằng cách gõ 'python' để truy cập trình thông dịch Python. Tiếp theo, nhập thư viện torch bằng cách gõ 'import torch'. Để kiểm tra xem CUDA có sẵn hay không, hãy sử dụng lệnh 'torch.cuda.is_available()'. Nếu nó trả về true, việc cài đặt của bạn đã thành công.

Câu hỏi thường gặp

Q: Các yêu cầu trước khi cài đặt PyTorch là gì?
A: Trước khi cài đặt PyTorch, hãy đảm bảo rằng Python và pip đã được cài đặt trên máy của bạn.
Q: Làm thế nào để tôi điều hướng đến trang cài đặt PyTorch?
A: Truy cập trang web chính thức của PyTorch và cuộn xuống để tìm các tùy chọn cài đặt.
Q: Làm thế nào để tôi chọn nền tảng tính toán phù hợp cho PyTorch?
A: Chọn phiên bản CUDA phù hợp cho card đồ họa Nvidia, ROCm cho AMD, hoặc CPU nếu không có GPU.
Q: Làm thế nào tôi có thể kiểm tra phiên bản CUDA đã cài đặt của mình?
A: Mở PowerShell và nhập lệnh 'nvcc --version'.
Q: Tôi cần làm gì để tải xuống và cài đặt CUDA?
A: Chọn hệ điều hành và kiến trúc của bạn trên trang tải xuống CUDA, chọn trình cài đặt 'exe local', và đảm bảo bạn có ít nhất 3.1 GB dung lượng trống.
Q: Làm thế nào để tôi hoàn tất cài đặt CUDA?
A: Khởi động trình cài đặt CUDA, đồng ý với thỏa thuận cấp phép, và tiếp tục với cài đặt nhanh.
Q: Làm thế nào tôi có thể cài đặt Visual Studio?
A: Tìm kiếm Visual Studio trực tuyến, tải xuống phiên bản Community, và chọn các gói bạn muốn cài đặt.
Q: Tôi nên làm gì sau khi cài đặt Visual Studio?
A: Quay lại trình cài đặt CUDA và chạy tệp thực thi một lần nữa để hoàn tất cài đặt CUDA.
Q: Làm thế nào để tôi cài đặt PyTorch?
A: Mở PowerShell và chạy lệnh pip phù hợp để cài đặt PyTorch.
Q: Làm thế nào tôi có thể xác minh rằng PyTorch đã được cài đặt đúng cách?
A: Chạy 'python' trong PowerShell, sau đó 'import torch', và kiểm tra xem CUDA có sẵn không với 'torch.cuda.is_available()'.

Chia sẻ đến

Trình duyệt vân tay chống phát hiện DICloak giữ cho việc quản lý nhiều tài khoản một cách an toàn và tránh bị cấm

Giúp việc vận hành nhiều tài khoản trở nên đơn giản, phát triển hiệu quả doanh nghiệp của bạn trên quy mô lớn với chi phi thấp

Bài viết liên quan