pix2pixHD尝试-安装相关内容
之前为了跑pix2pix,安装过python、cuda等相关内容(参见我之前的文章)
源代码:https://github.com/NVIDIA/pix2pixHD
环境配置
安装cuda和pytorch
安装dominate
安装/升级其他Python包
pytorch的安装和torchvision。注意需要对应python和cuda的版本下载,否则会出现版本的对应错误。
需要升级python的一些东西,比如pillow、numpy。
对应显卡的版本,cuda版本应该与之对应,否则会出现警告:Found GPU0 GeForce GTX 1660 SUPER which requires CUDA_VERSION >= 9000 for optimal performance and fast startup time, but your PyTorch was compiled with CUDA_VERSION 8000.
在win系统运行的话参考文章:https://blog.csdn.net/guoqiangszu/article/details/83302619