最近实习工作需要多个环境的pytorch与cuda，目前所需为pytorch-0.4.1(cuda9.0/cudnn7.5.1/python3.6)与pytorch-0.4.1(cuda8.0/cudnn5.1/python2.7)，pytorch和python版本可以使用virtualenv控制，cuda版本需要安装多个版本cuda，然后生成要使用cuda版本的软链接，将软链接地址加入到环境变量中。
删除原软链接 sudo rm -rf /usr/local/cuda 建立新的软链接 sudo ln -s /usr/local/cuda-8.0 /usr/lcoal/cuda可以在/usr/local目录下使用stat cuda查看cuda对应的软链接，也可以使用nvcc -V查看当前cuda版本
修改环境变量 sudo gedit ~/.bashrcexport PATH=/usr/local/cuda/bin:$PATHexport LD_LIBRARY_PATH=/usr/local/cuda/lib64$LD_LIBRARY_PATHexport CUDA_HOME=/usr/local/cuda/etc/profile文件也需要修改
sudo gedit /etc/profileexport PATH=/usr/local/cuda/bin:$PATHexport LD_LIBRARY_PATH=/usr/local/cuda/lib64$LD_LIBRARY_PATHexport CUDA_HOME=/usr/local/cuda 使用命令查看当前cuda软链接地址 ls -l cuda