conda环境中同时使用tf与pytorch报错 InvalidArgumentError: assertion failed: [0] [Op:Assert]

it2023-04-17  76

环境:tf-gpu=2.3.0、torch=1.5.0、cuda=10.1

解决办法:先import tensorflow,再import torch

改变前(报错):

import torch import tensorflow

改变后(不报错):

import tensorflow import torch

 

 

 

最新回复(0)