安装python3
https://www.python.org/downloads/windows/
https://www.python.org/ftp/python/3.9.13/python-3.9.13-amd64.exe
以Tesla T4显卡为例
安装显卡驱动
https://www.nvidia.cn/content/DriverDownloads/confirmation.php?url=/tesla/538.33/538.33-data-center-tesla-desktop-win10-win11-64bit-dch-international.exe&lang=cn&type=Data%20Center%20/%20Tesla
安装cuda
CUDA Toolkit: https://developer.nvidia.com/cuda-11.1.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
安装cudnn
安装说明:https://docs.nvidia.com/deeplearning/cudnn/installation/windows.html
下载页面:
https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/windows-x86_64/
https://developer.nvidia.com/cudnn-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exe_local
cudnn的python包
pip install nvidia-cudnn-cu11
或者:
pip install -i https://mirrors.cloud.tencent.com/pypi/simple/ nvidia-cudnn-cu11==8.9.5.29