问题:最近笔者在训练神经网络,数据量虽小,但有一处神经网络训练及其费时(可能我不太会调参或者改写),于是我想要启用GPU来跑,笔者GPU是Nvidia的GTX1650,一块很古早的显卡了,这块显卡我在Nvidia官网甚至查不到训练神经网络的性能(1650ti就有),然后一顿配啊......
出错:版本老是对不上,我一开始的conda虚拟环境一堆乱七八糟的包,然后自己下了cuda和cudnn,发现还是用不了gpu,最后自己新建了环境,一个一个装包了。所以出错在哪呢?cuda和cudnn与tensorflow版本不对付;tensorflow与python版本不对付;numpy与python版本不对付;numpy与tensorflow版本不对付。总之就是更新了跟这个不对付,降版本又跟那个不对付。
解决:直接看版本信息吧,其实知道哪些版本对付,直接敲指令也就行了。
GPU version:NVIDIA GeForce GTX1650
python=3.9
tensorflow=2.10.0
cuda=11.2
cudnn=8.1
numpy=1.24.4

对了,我最近在搞对抗样本生成,其中有一个导入包叫art,他其实跟另一个art包没关系,它其实是adversarial-robustness-toolbox,下载包的时候要注意,否则下错包的话,参考代码的所需的一些文件是导入不进来的。
心血来潮作此记录,主要是给自己留个印象,欢迎指正。
2491

被折叠的 条评论
为什么被折叠?



