赞
踩
有时候复现他人的代码需要配置各种不同的环境,自己的电脑因为只安装一个版本的cuda,所以无法满足各个程序的环境配置。
因此,我们可以在Pycharm的虚拟环境下安装cuda,每个工程配置一个自己的虚拟环境和对应的cuda版本,这样就可以任意切换cuda版本,并且不会影响外部环境。
首先需要在电脑上安装Anaconda,Anaconda的安装方式及版本对应可参考我的另一篇博客:下载安装Anaconda(对应Python版本)
然后需要在pycharm下设置虚拟环境,点击File-Settings-Python interpreter
然后如图点击Add…
然后选择conda的虚拟环境安装方式,选择python版本。
在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。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。