tensorflow2.X+2080显卡环境配置

anaconda+pycharm

python==3.6.13

keras==2.1.6

cuda==11.7(需要看显卡支持的最低版本,2080的是11.5)

cudnn==8.4.0(需要和cuda版本对应相匹配)

tensorflow-gpu==2.4.0(需要和cudnn+cuda相对应匹配,最低版本2.4.0)

如果需要使用tensorflow-gpu 1.x版本的话,需要降低cuda和cudnn的版本10+7.4

下载测试代码测试

如果出现测试module 'tensorflow' has no attribute 'constant'错误,可能是tensorflow-gpu版本低了

如果出现Could not load dynamic library ‘cusolver64_10.dll‘; dlerror: cusolver64_10.dll not found错误,GPU  fasle的话需要进入安装cuda的库里面,搜索cusolver64_11.dll,将其修改为cusolver64_10.dll

本次记录是为了跑Maskrcnn 模型,如果在训练maskrcnn时,出现

module 'tensorflow' has no attribute 'placeholder'错误,主要是因为这个版本的maskrcnn的代码使用的tensorflow1.X里的函数,这些函数在tensorflow2.x版本是没有的

测试结果如下

安装包如下

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
TensorFlow 1.1是一个深度学习框架,而GeForce RTX 3060是一款显卡。要在Windows操作系统上使用TensorFlow 1.1与GeForce RTX 3060一起工作,需要进行一些配置和安装步骤。 首先,确保已在计算机上安装了适当的显卡驱动程序。可以通过NVIDIA官方网站下载并安装最新的显卡驱动程序,以确保与GeForce RTX 3060兼容。 接下来,可以通过两种方式安装TensorFlow 1.1。一种是使用pip包管理器,打开命令提示符窗口并输入以下命令: ``` pip install tensorflow==1.1 ``` 此命令将从Python Package Index(PyPI)下载并安装TensorFlow 1.1。 另一种方式是创建一个虚拟环境来安装TensorFlow。可以使用Anaconda或者Python的venv模块创建虚拟环境,然后在虚拟环境中使用pip安装TensorFlow 1.1。 安装完成后,可以通过导入TensorFlow并编写代码来验证TensorFlow 1.1在Windows上与GeForce RTX 3060一起工作。可以使用以下代码作为一个简单的例子: ``` import tensorflow as tf # 创建一个简单的神经网络模型 model = tf.keras.models.Sequential([ tf.keras.layers.Dense(64, input_shape=(10,), activation='relu'), tf.keras.layers.Dense(1, activation='sigmoid') ]) # 编译模型 model.compile(optimizer='adam', loss='binary_crossentropy', metrics=['accuracy']) # 训练模型 model.fit(x_train, y_train, epochs=10, batch_size=32) # 预测新的数据 predictions = model.predict(x_test) ``` 以上是使用TensorFlow 1.1与GeForce RTX 3060在Windows上进行深度学习的基本步骤。但需要注意的是,TensorFlow 1.1是一个较旧的版本,可能与一些最新的功能和库不兼容。为了获得更好的兼容性和性能,建议升级到TensorFlow的最新版本

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值