https://zhuanlan.zhihu.com/p/91334380
不同的pytorch版本在用conda安装时,会安装不同的cudatoolkit版本, 其主要包含应用程序在使用 CUDA 相关的功能时所依赖的动态链接库,不会安装驱动程序。与系统cuda不同(usr/local/cuda), Nvidia 官方提供的 CUDA Toolkit 是一个完整的工具安装包,其中提供了 Nvidia 驱动程序、开发 CUDA 程序相关的开发工具包等可供安装的选项。
有时候pytorch的cudatoolkit版本与系统cuda版本不同时, 没有影响可以使用; 有时候安装一些库时会报错, 比如torch-geometric
本文讨论了PyTorch不同版本与CUDA版本之间的兼容性问题,解释了PyTorch安装时如何选择对应的CUDAToolkit版本,并探讨了当PyTorch的CUDAToolkit版本与系统CUDA版本不一致时可能遇到的问题。
1万+

被折叠的 条评论
为什么被折叠?



