一、安装cuda
cuda下载(版本12.0.0)
自定义安装可以只安装Cuda Runtime
默认安装目录:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.0
二、安装cuDNN
cuDNN下载(版本8.8.1.3)
将解压的cuDNN压缩包内的三个文件夹(bin、lib、include)复制到CUDA安装目录下
二、torch安装
pip install torch==1.13.1 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117
三、对应版本onnxruntime安装(若需要)
pip install onnxruntime-gpu1.17.1
pip install onnxruntime-gpu1.17.0 --index-url=https://pkgs.dev.azure.com/onnxruntime/onnxruntime/_packaging/onnxruntime-cuda-12/pypi/simple