pytorch和torch这两个框架有些地方不一样
cuda与pytorch版本对应网站:
https://pytorch.org/get-started/previous-versions/
cudatoolkit就是cuda
查看cudnn版本:
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
(一般下载pytorch的时候就会自动下载)
pytorch与cuda、cudnn版本对应问题
最新推荐文章于 2024-08-16 09:41:32 发布