美文网首页
AnimeGAN人脸动漫化

AnimeGAN人脸动漫化

作者: 李霖弢 | 来源:发表于2023-06-14 21:00 被阅读0次
在线预览
git仓库

环境配置

nvidia-smi

nvidia-smi是N卡驱动附带的命令行指令,可以查看当前的GPU信息:
其中 Driver Version是当前NVIDIA显卡驱动版本,CUDA Version是该驱动可用的CUDA版本上限

CUDA

CUDA是由NVIDIA推出的通用并行计算架构,使GPU能够解决图像以外的计算问题。
已安装的CUDA会附带nvcc -V命令行指令,用于查看版本

pytorch

pytorch是基于torch的开源机器学习库,支持跨平台:Windows(CUDA,CPU)、MacOS(CPU)、Linux(CUDA,ROCm,CPU)
pytorch的包名就叫torch
pytorch和CUDA版本对应关系可见官网链接:https://pytorch.org/get-started/previous-versions/。通常高版本的pytorch可以兼容低版本的CUDA。

  • 下载安装
    官网首页生成下载链接,或者在pytorch下载列表中找到符合的版本下载后再手动通过pip载入。其中cu为cuda版本,torch为torch的版本,cp为python版本。
    注意直接通过pip install torch安装的默认是CPU版。可通过import torch; print(torch.__version__)确认是CPU还是CUDA版。
    可通过import torch; print(torch.cuda.is_available())确认当前已安装的torch版本是否能与本机CUDA正常协作。

相关文章

网友评论

      本文标题:AnimeGAN人脸动漫化

      本文链接:https://www.haomeiwen.com/subject/ojjcydtx.html