GeForce GTX1650启用GPU

部署运行你感兴趣的模型镜像

问题:最近笔者在训练神经网络,数据量虽小,但有一处神经网络训练及其费时(可能我不太会调参或者改写),于是我想要启用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,下载包的时候要注意,否则下错包的话,参考代码的所需的一些文件是导入不进来的。

心血来潮作此记录,主要是给自己留个印象,欢迎指正。

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch 2.5

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值