估计你已经被深度学习环境的安装和配置整得焦头烂额了,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/