conda create -n py38 python=3.8
conda search cudatoolkit --info
conda search cudnn --info
(3)pip方式安装
pip install -r requirements.txt
requirements.txt内容:
–extra-index-url https://pypi.ngc.nvidia.com
nvidia-cuda-runtime-cu12
nvidia-cuda-cupti-cu12
nvidia-cuda-nvcc-cu12
nvidia-nvml-dev-cu12
nvidia-cuda-nvrtc-cu12
nvidia-nvtx-cu12
nvidia-cuda-sanitizer-api-cu12
nvidia-cublas-cu12
nvidia-cufft-cu12
nvidia-curand-cu12
nvidia-cusolver-cu12
nvidia-cusparse-cu12
nvidia-npp-cu12
nvidia-nvjpeg-cu12
(4)官网安装,3种安装方式,我用的deb(local)
官网链接cuda12.1安装
安装成功后,查看版本: