一行命令解决 深度学习环境 TensorFlow PyTorch Keras安装 cuda cudnn 配置

估计你已经被深度学习环境的安装和配置整得焦头烂额了,cuda和cudnn的安装配置真的好麻烦,下面我将告诉你只要用一行代码就可以解决这个问题。
不过首先,你得下载安装anaconda环境,而不是使用纯Python软件。
然后创建一个你想要运行深度学习的环境。
一般使用过Python的人,以上两步都已经完成,新手的话可以自行查阅如何解决,很简单。

最后。。。。。。

进入终端,进入自己的环境。。。。。。。

TensorFlow:

conda install tensorflow-gpu

PyTorch:

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

anaconda自动为你选择所有所需下载和更新的包,包括cuda和cudnn,你只需要等待下载安装完成即可。

大功告成!

提示

pytorch的安装命令可以自行查找自己所需的版本:
https://pytorch.org/

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章