机器之心报道
参与:杜伟、一鸣
TensorFlow2.1的更新,能够让弃坑的用户回心转意吗?
去年10月,谷歌才发布了TensorFlow2.0正式版。时隔三个月后,昨日官方发布了TensorFlow2.1,本次版本更新带了了多项新特性、功能改进和bug修复。
从本次更新的日志来看,TensorFlow2.1将成为最后一个支持Python2的版本了。同时,本次更新的重点是增加了对TPU的多项支持,而tf.keras和tf.data这两个常用的API也得到了很多新的更新。据悉,TensorFlow2.1的CUDA版本为10.1,cuDNN版本为7.6。
在对操作系统的支持上,Windows和Linux系统的TensorFlowpip版本默认支持GPU。也就是说,如果使用pipinstalltensorflow,则版本默认为是gpu版本(原始的tensorflow-gpu版本依然存在)。当然,不管有没有英伟达版本的GPU,tensorflow依然能够运行。如果需要使用CPU版本,用户的安装命令应该为:pipinstalltensorflow-cpu。项目