开题:gtx3090 cuda 11.0 tensorflow 2.4 2.5 这些版本我没配置成功。
从泰坦显卡转战3090,今天下午配置环境。先是安装了TF2.2.没想到只能调用CPU训练。于是我安装 tf2.4 以及没发布的tf2.5.
均未果。
以下是我尝试的一些方法:
首先安装官网给出的步骤,安装TF2.4
pip install tensorflow==2.4.0
跑代码,结果出现如下报错:
Your CUDA software stack is old. We fallback to the NVIDIA driver for some compilation. Update your CUDA version to get the best performance. The ptxas error was: ptxas fatal : Value ‘sm_86’ is not defined for option ‘gpu-name’
github有同样问题issure
这里的情况是,gpu我看已经占了,但是不断打印 上述信息。
又参考了如下的文章
安装TF2.4(这个都不跑代码)
https://blog.youkuaiyun.com/weixin_45114252/article/details/109463521
转战TF2.5
https://blog.youkuaiyun.com/weixin_41194129/article/details/114238772
安装了未发布的 TF2.5
https://libraries.io/pypi/tf-nightly-gpu
未果。
还有这篇文章:
https://blog.youkuaiyun.com/weixin_43233764/article/details/111536819
他成功了,我未成功。
最终解决方法 ,clone 师兄已经安装号的TF2.5虚拟环境。
版本如下。
CUDA 版本号如下:
我的时间就是这么没了。希望上述链接及我的探索对你有帮助。