CUDA Python 環境建置

Python 版本依據專案需要選擇,推薦使用 3.12,但目前最高相容性依然是 3.10。

  1. CUDA Toolkits ( Download from https://developer.nvidia.com/cuda-toolkit )
    Check version> nvcc –version
    Check version> nvidia-smi
  2. Pytorch Using CUDA 12.4 ( Choose matched version on https://pytorch.org )
    pip3 install torch torchvision torchaudio –index-url https://download.pytorch.org/whl/cu124
  3. Tensorflow
    pip3 install tensorflow

留言

Leave a comment

This site uses Akismet to reduce spam. Learn how your comment data is processed.