Linux安装conda
只需要安装miniconda就行(Miniconda3仅包含conda、Python和一些必要的包以及少量其他有用的包)
见:https://docs.conda.io/projects/conda/en/stable/user-guide/install/linux.html
启动环境
conda create -n ocr python=3.8
# 激活环境
conda activate ocr
# 退出环境
conda deactivate
conda环境的安装cuda
可以用conda来做cuda和cudnn版本隔离,实现一个项目对应不同的cuda和cudnn版本。
注意:conda下的cuda还是用宿主机的驱动,受到宿主机驱动支持的最高cuda版本的限制
conda install cudatoolkit==11.8 -c conda-forge
conda install cudnn==8.8.0.121 -c conda-forge
可以用search
来查看目前所有的cuda版本,注意用conda-forge
通道,版本最多最全
conda search cudnn -c conda-forge
注意: 如果在 Conda 环境中没有安装 CUDA Toolkit(cudatoolkit),而你直接执行 conda install cudnn 来安装 cuDNN,cuDNN 文件会被安装到本地机器上已经安装的 CUDA 目录中。这意味着 Conda 会尝试将 cuDNN 安装到系统中已有的 CUDA 安装目录下,而不是安装到 Conda 环境中的 CUDA 目录。
因此,在这种情况下,cuDNN 文件会直接安装到本地机器的 CUDA 目录中,可能会覆盖或更新已有的 cuDNN 文件,需要注意这一点以避免与已有的 CUDA 安装发生冲突。如果你希望将 cuDNN 安装到 Conda 环境中的 CUDA 目录下,建议先确保在 Conda 环境中安装了对应版本的 CUDA Toolkit,然后再安装 cuDNN。
如果要安装torch,官方文档提供的命令会包含安装cuda,不过经常找不到这个版本的cuda和torch无法安装。建议还是自己从conda-forge安装好cuda后,用pip来安装torch
评论区