已知pytorch版本为1.4.0,CUDA版本为10.1。requirement中要求tensorflow-gpu版本为1.4.0,对应cuda8。这个该如何解决?是下载更高版本的TensorFlow吗?
3条回答 默认 最新
爱晚乏客游 2022-07-04 15:01关注0.cuda可以多环境共存,只要设置好环境变量即可。
1.升级tf到cuda10.1的版本,应该是1.13还是1.12来着,具体你可以去官网查一下对应关系
2.如果你真的要tf1.4的同时也要torch1.4,那么用anaconda新建环境,这两个环境分开安装,torch1.4+cuda10.1同一个环境,tf1.4+cuda8.0一个环境。本回答被题主选为最佳回答 , 对您是否有帮助呢?解决 无用评论 打赏 举报