通过pip安装cuda

发布于:2023-03-12 ⋅ 阅读:(62) ⋅ 点赞:(0)

五六年前深度学习还是个新鲜事的时候,linux下显卡驱动、CUDA的很容易把小白折磨的非常痛苦,以至于当时还有一个叫manjaro的发行版,因为驱动安装简单流行。老黄也意识到了这个问题,增加了很多新的安装方式。

最近发现了使用pip安装CUDA的方式。这种方法只安装一些必要的运行时,没有开发工具。好处是可以配合python的虚拟环境,无痛管理环境,windows/linux都能适用。但如果想在python以外使用,需要一些额外设置。

方法非常简单,只需要先编写一个requirements.txt,然后使用pip安装。

requirements.txt文件:

--extra-index-url https://pypi.ngc.nvidia.com

nvidia-cuda-runtime-cu11
nvidia-cuda-cupti-cu11
nvidia-cuda-nvcc-cu11
nvidia-nvml-dev-cu11
nvidia-cuda-nvrtc-cu11
nvidia-nvtx-cu11
nvidia-cuda-sanitizer-api-cu11
nvidia-cublas-cu11
nvidia-cufft-cu11
nvidia-curand-cu11
nvidia-cusolver-cu11
nvidia-cusparse-cu11
nvidia-npp-cu11
nvidia-nvjpeg-cu11

文件中的cu11代表CUDA 11,具体的11.X会自动选取。如果需要手动选取特定版本,应该直接写成cu11x就行了。

安装指令:

pip install -r requirements.txt

一般来说,国内会自动解析到英伟达的国内下载站(developer.download.nvidia.cn),所以速度非常快。


网站公告

今日签到

点亮在社区的每一天
去签到