在源码安装pytorch时,我的cuda的版本时11.4,所以按照官方的安装指南是需要magma-cuda114,而此时magma-cuda114还没发布。
所以我的方法是
卸载驱动、CUDA和CUDNN(具体自己百度)。然后选择对应cuda版本为11.3的显卡驱动,然后安装cuda11.3和相应的CUDNN。
说明:显卡的驱动可以用较低版本的。

在源码安装pytorch时,我的cuda的版本时11.4,所以按照官方的安装指南是需要magma-cuda114,而此时magma-cuda114还没发布。
所以我的方法是
卸载驱动、CUDA和CUDNN(具体自己百度)。然后选择对应cuda版本为11.3的显卡驱动,然后安装cuda11.3和相应的CUDNN。
说明:显卡的驱动可以用较低版本的。