最新消息:

Python

moviepy用GPU加速的方法 – torch

2024-07-10 768浏览 0评论

以下修改在moviepy 1.0.3,CUDA为12.1.1,Pytorch为Windows Cuda12.1的环境下测试通过。确实比CPU生成有提升,大约40%。 moviepy默认的安装路径为: C:\Users\Administrator\Ap...

检测CUDA、cuDNN、Pytorch是否可用

2024-07-10 277浏览 0评论

将下面的代码保存成.py文件,运行即可得到结果。 import torch print('CUDA版本:',torch.version.cuda) print('Pytorch版本:',torch.__version__) print('显卡是否可用...