Python 版本依據專案需要選擇,推薦使用 3.12,但目前最高相容性依然是 3.10。
- CUDA Toolkits ( Download from https://developer.nvidia.com/cuda-toolkit )
Check version> nvcc –version
Check version> nvidia-smi - 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 - Tensorflow
pip3 install tensorflow

Leave a comment