AnimeGAN人脸动漫化

2023-06-14  本文已影响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。

上一篇 下一篇

猜你喜欢

热点阅读