已知pytorch版本为1.4.0,CUDA版本为10.1。requirement中要求tensorflow-gpu版本为1.4.0,对应cuda8。这个该如何解决?是下载更高版本的TensorFlow吗?
0.cuda可以多环境共存,只要设置好环境变量即可。
1.升级tf到cuda10.1的版本,应该是1.13还是1.12来着,具体你可以去官网查一下对应关系
2.如果你真的要tf1.4的同时也要torch1.4,那么用anaconda新建环境,这两个环境分开安装,torch1.4+cuda10.1同一个环境,tf1.4+cuda8.0一个环境。
那就将ensorflow-gpu版本升为1.4.0呗
请看👉 :python多版本测试开发