python虚拟环境深度学习环境配置Linux

python虚拟环境深度学习环境配置

  • 创建python3.9的虚拟环境
  • 安装cuda以及cudnn
    • 查找合适的cuda版本
    • 查找合适的torch版本(会同时安装cuda)
    • 检查是否安装成功
    • cudnn环境不匹配修改
      • 编辑activate.d/env_vars.sh,注意将your_name换成你的环境名字,注意=前后不能有空格哦
      • 编辑deactivate.d/env_vars.sh,这样deactivate的时候,会恢复设置
      • 保存之后重启虚拟环境
      • 再次检查是否安装成功
      • 参考连接

创建python3.9的虚拟环境

# 在合适的地方打开终端

# 用python的venv创建,二选一即可
sudo apt install python3.9-venv #使用以下命令来安装 python3-venv 包,这将包括创建虚拟环境所需的所有依赖,包括 ensurepip
python3.9 -m venv env #创建了一个名为env的虚拟环境,python的版本为3.9

# 用conda进行创建,二选一即可
conda create -n env python=3.9

#激活名为env的虚拟环境
conda activate env

安装cuda以及cudnn

查找合适的cuda版本

nvidia-smi #查看显卡信息

python虚拟环境深度学习环境配置Linux_第1张图片
我的推荐版本为12.0的cuda,但是为了与tensorrt的版本配套,选择11.8的cuda

查找合适的torch版本(会同时安装cuda)

torch链接地址:https://pytorch.org/get-started/previous-versions/

你可能感兴趣的:(python,深度学习,linux)