在运行pix2pix时遇到了上面的问题:
配置:cuda10.0
cudnn7.4.2
显卡RTX2080ti
安装pytorch(利用pip安装1.1.0,1.0.1,1.5.0都安装了一遍,且相对应的torchvision版本也安装了一遍)都会出现以上问题。
原因是显卡用的RTX 2080Ti,CUDA就要装10以上,这个时候pytorch不能直接用pip装,按照下面指令在终端里面安装。
pip3 install https://download.pytorch.org/whl/cu100/torch-1.0.0-cp36-cp36m-linux_x86_64.whl