Pytorch 使用特定的 cuda 版本

（仅作为个人学习所用 原文链接Pytorch 使用不同版本的 cuda - yhjoker - 博客园）

从 Pytorch 确定使用的 cuda 版本的流程来看，想要指定 Pytorch 使用的 cuda 版本，主要有两种方法，第一种是修改软链接 /usr/local/cuda 所指向的 cuda 安装目录( 若不存在则新建 )，第二种是通过设置环境变量 CUDA_HOME 指向所需使用的 cuda 版本的安装目录。除此之外，还建议将对应 cuda 目录中的可执行文件目录( 形如/home/test/cuda-10.1/bin )加入环境变量 PATH 中。

对于第一种方法，由于 /usr/ 和 /usr/local/ 目录下的文件均为 root 用户所管理，故而普通用户无法对其进行修改。对于具备了 root 权限的用户而言，在安装有多版本 cuda 的 Linux 系统上，只需切换 /usr/local/cuda 所指向的 cuda 目录，让其指向所需的 cuda 版本的安装位置，即可让 Pytorch 在运行时使用指定版本的 cuda 运行程序。修改软链接的方法如下命令所示，命令删除原有的软链接，并新建指向新路径的软链接。

　　sudo rm -rf /usr/local/cuda        　　　//删除软链接，注意是 /usr/local/cuda 而不是 /usr/local/cuda/,前者仅删除软链接，而后者会删除软链接所指向的目录的所有内容，操作请小心
　　sudo ln -s cuda_path /usr/local/cuda    //创建名为 /usr/local/cuda 的软链接，其指向 cuda_path 所指定的 cuda 安装目录

或者直接强制修改原始的软链接

sudo ln -sf cuda_path /usr/local/cuda    //修改或创建软链接 /usr/local/cuda 使其指向指定版本的 cuda 目录

对于非 root 用户而言，主要通过第二种方法进行设置。若想要指定 Pytorch 使用的 cuda 版本，则首先需要设置 CUDA_HOME 环境变量，之后在 PATH 中加入指定 cuda 版本的可执行目录，也就时 cuda_path/bin/ 目录。完成设置后，运行 Pytorch 时所使用的即为对应的 cuda 版本。

参考：Pytorch 使用不同版本的 cuda - yhjoker - 博客园