最近做创新实验是做有关图像识别的内容,所以需要装tensorflow。
为了减少不必要的麻烦,我把我之前下的tensorflow版本删了,再重新装的这个版本。
删除tensorflow可以使用下面这行命令,简单删除:
pip uninstall tensorflow
使用这条命令,你可以在PyCharm下、cmd窗口上直接输入,至于其他的方法,大家可以再另外找找。
刚开始我不知道直接在Anoconda Prompt 下输入命令直接安装,所以走了点弯路。
当时装CDUA和cuDNN参考的是这篇博文,我当时装的CUDA版本是10.2版本,然后cuDNN装的是对应版本,见下图
大家有需要的话,可以参考一下下面这篇文章,我用文章的方法也成功装上了CUDA和cuDNN,如果装不同版本的话,需要注意一下在添加路径时,将10.1改成你所装的版本名。
博主:Bubble_water
windows10+nvidia驱动+cuda10.1+cudnn安装教程
重申一下,我原先在电脑装的CUDA10.2 版本并没有删除。但就目前运行的情况来说没有问题。可能是因为我是在新建的TF2.1环境下再装的CUDN10.1+cuDNN7.6
版本,所以可能没有冲突。(ps:这个我不太了解,这个只是我的推测哈🧡,但是装上CUDA10.2以后,我也没有运行过CUDA10.2,所以不太清楚什么情况)
另外附上查看电脑GPU版本的方法:
1.打开控制面板→硬件和声音
鼠标右击NVIDIA控制面板,点击打开,就可以看到自己电脑的GPU版本了。
我是按照上面那篇博文,安装的CUDA10.2版本和cuDNN v7.6.5版本。
但是我昨天下午安装好后,发现还需要配置专门的环境才可以在PyCharm中使用GPU运行tensorflow,不然的话,会报错。索性我昨晚就重新看教程安装了一次,今早上终于把它给安装好了,由于我在安装的过程中没有截图,所以可能具体的安装情况不能给大家展示,下面我就附上操作步骤和简单说明。
首先,有几大步骤,分别是:
- Anaconda 安装
- conda install cudatoolkit=10.1
- conda install cudnn=7.6
- pip install tensorflow==2.1
- PyCharm 安装 <