pytorch gpu安装

无由頁2023-10-03 16:07

cuda

https://blog.csdn.net/qq_51570094/article/details/124148671

https://blog.csdn.net/zxdd2018/article/details/127705627

cudnn

https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#installlinux-tar

更改cudnn

保证文件目录中只有一个解压后的cudnn文件夹

复制代码

sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include 
sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64 
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

pytorch版本匹配问题

nvcc --version和nvidia-smi的版本可能會不一樣，前者是runtime api對應的版本，後者是driver api的版本。

pytorch版本通常需要和runtime的版本匹配，driver的版本可以向下兼容。