安装TensorFlow-gpu的注意点

直接pip install tensorflow,安装到的是tensorflow-gpu 1.14.0,对应的CUDA版本是10.0,我为啥知道?本以为对应的是最新的CUDA10.1结果安好就悲剧了,还好软件提示:无法找到cudart64_100,请下载CUDA10.0,不过后面给的连接比较坑爹,给的是cuda9.0的,老实的我又中招了,下载了cuda9.0还是用不了。。。后来在下面找到了10.0
相关文章
相关标签/搜索