在Pycharm下安装多个版本的cuda

有时候复现他人的代码需要配置各种不同的环境,自己的电脑因为只安装一个版本的cuda,所以无法满足各个程序的环境配置。
因此,我们可以在Pycharm的虚拟环境下安装cuda,每个工程配置一个自己的虚拟环境和对应的cuda版本,这样就可以任意切换cuda版本,并且不会影响外部环境。

1. 安装Anaconda

首先需要在电脑上安装Anaconda,Anaconda的安装方式及版本对应可参考我的另一篇博客:下载安装Anaconda(对应Python版本)

2. 在Pycharm下设置虚拟环境

然后需要在pycharm下设置虚拟环境,点击File-Settings-Python interpreter
在这里插入图片描述然后如图点击Add…
在这里插入图片描述
然后选择conda的虚拟环境安装方式,选择python版本。

3. 安装cuda和cudnn

在pycharm创建的conda虚拟环境的终端下安装cuda和cudnn。安装方式选择conda的安装方式,资源从清华源下载。

#install cuda
conda install cudatoolkit=10.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
#install cudnn
conda install cudnn=7.6.5 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/
#install package if cannot directly install cudnn (opt)
conda install --use-local cudnn7.6.5.tar.bz2

到此,就相当于在虚拟环境下安装了cuda。

猜你喜欢

转载自blog.csdn.net/gaoqing_dream163/article/details/110532133