pix2pixHD尝试-安装相关内容

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

  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值