由于使用深度學習框架的不同,有的時候我們需要切換cudnn環(huán)境。比起在系統(tǒng)中安裝多個cudnn版本,更便捷的方法是通過在python環(huán)境下安裝cudnn工具,這樣不同的cudnn環(huán)境就可以用python的包管理器(如conda等)管理,使用起來很方便。
最常用的方式是在conda下,通過安裝不同版本的cudatoolkit來滿足要求。文章來源:http://www.zghlxwxcb.cn/news/detail-664878.html
conda install cudatoolkit
然而有的時候我們用的包管理器不是cuda,或者我們用的python包鏡像不支持cuda,這時只能用pip. 以cuda11為例,此時可以使用以下指令安裝需要的cudnn工具;注意選擇自己需要的版本號。文章來源地址http://www.zghlxwxcb.cn/news/detail-664878.html
pip install nvidia-cudnn-cu11
pip install nvidia-cuda-runtime-cu11
pip install nvidia-cublas-cu11
pip install nvidia-cuda-nvrtc-cu11
到了這里,關于Python:conda install cudatoolkit的備選方案:pip install nvidia-cudnn-*:的文章就介紹完了。如果您還想了解更多內容,請在右上角搜索TOY模板網以前的文章或繼續(xù)瀏覽下面的相關文章,希望大家以后多多支持TOY模板網!