在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。

  • 1
    点赞
  • 15
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值