高版本cuda使用低版本cudatoolkit

  1. 换科大pip源

pip install -i https://mirrors.ustc.edu.cn/pypi/web/simple pip -U
pip config set global.index-url https://mirrors.ustc.edu.cn/pypi/web/simple

临时使用
> pip install -i https://mirrors.ustc.edu.cn/pypi/web/simple package
  1. 安装cudatoolkit

  2. 安装cuda10.2:

    conda install cudatoolkit=10.2

  3. 安装cudnn :

    conda install cudnn(可以不加版本,自动匹配cuda)

    3090不支持11以下版本的cuda,需要装11版本的cuda

你可能感兴趣的:(python,深度学习,开发语言)