当前位置:   article > 正文

在Pycharm下安装多个版本的cuda_pycharm使用不同cuda

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
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6

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

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/凡人多烦事01/article/detail/202151
推荐阅读
相关标签
  

闽ICP备14008679号