安裝環境及版本
顯卡:NVIDIA GeForce RTX 3050 Laptop GPU
OS:Windows 10 Home, 64-bit
Python 3.8.*(pytorch目前最高只支援到3.8; 建議使用 anaconda 以利試誤)
TensorFlow 2.8
CUDA ToolKit 11.3(pytorch目前最高只支援到11.3; 建議使用 anaconda 以利試誤)
cuDNN 8.3
安裝 CUDA Toolkit 11.3
1. NVIDIA官網下載CUDA Toolkit 11.3 安裝檔,並且執行。
2. 查看CUDA安裝版本
nvcc --version
安裝cuDNN 8.3
1. NVIDIA官網下載cuDNN (需免費註冊帳號)的壓縮檔 (抓安裝檔執行後仍需進行以下步驟)。
2. 將壓縮檔內bin、include、lib子資料夾複製到下列資料夾:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\
(這一步是最令人費解的,為何不讓安裝檔直接找到應該安裝的位置呢? 總之詳參 3.3. Installing On Windows )
安裝 TensorFlow & pytorch
1. 安裝指令
pip install tensorflow
2. 查看TensorFlow安裝版本
pip show tensorflow
3. 安裝 pytorch
conda install pytorch torchvision torchaudio PySoundFile cudatoolkit=11.3 -c pytorch
4. 查看 pytorch 安裝版本
python -c "import torch ; print(torch.__version__)"
檢測是否完成安裝
1. cmd 下執行
python -c "from tensorflow.python.client import device_lib;print(device_lib.list_local_devices())"
若出現 gpu 等字樣代表安裝成功
2. 或 cmd 下執行 python
>>import tensorflow as tf
>>import os
>>os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
>>print('GPU', tf.config.list_physical_devices('GPU'))
若出現如下訊息,代表安裝成功
GPU [PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]
(亦可參 Win10 安裝 CUDA、cuDNN 教學 有較詳細講解),作者是大正妹)