【环境配置】多cuda版本多pytorch版本配置

最新推荐文章于 2025-05-14 15:30:36 发布

原创

最新推荐文章于 2025-05-14 15:30:36 发布 · 1.1k 阅读

10 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #深度学习

【环境配置】多cuda版本多pytorch版本使用。

CUDA 11.3+pytorch 1.12.1

1.安装CUDA11.3

2.安装cudnn版本。注意，这里使用安装到CUDA文件夹的方式。
安装cudnn 到cuda-11.3文件夹：

1.下载cudnn的压缩包。在 cudnn下载中选择for 11.x。下载时选择：Local Installer for Linux x86_64 (Tar)。
2.解压：
tar cudnn-10.1-linux-x64-v8.0.4.30.tgz #解压
3.拷贝到cuda文件夹
sudo cp cuda/include/cudnn.h /usr/local/cuda-11.3/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.3/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda-11.3/lib64/libcudnn*

3.安装pytorch 1.12.1。
写入cuda路径到YOUR_USER/bashsrc：