当前位置:   article > 正文

conda虚拟环境 + 多cuda + pytorch_gpu + tensorflow_gpu配置_cuda虚拟环境

cuda虚拟环境

一、conda虚拟环境

查看现有环境:conda env list
创建环境:conda create -n 环境名 python=3.8 (python版本,可更改)
激活:conda activate 环境名
退出激活:conda deactivate 
删除环境:conda remove -n 环境名 --all
  • 1
  • 2
  • 3
  • 4
  • 5

将环境配置到jupyter notebook上,将环境加载进jupyter中

pip install ipykernel  【pip install -i https://pypi.douban.com/simple ipykernel (使用豆瓣源速度快)】
python -m ipykernel install --name 环境名
  • 1
  • 2

二、cuda安装

1、查看显卡支持的cuda版本

打开 ‘ NVIDIA控制面板 ’

在这里插入图片描述

点击 ‘ 帮助 ’ —> ‘ 系统信息 ’

在这里插入图片描述

点击 ‘ 组件 ’

在这里插入图片描述

可以向下兼容;例如我的:<= 12.1都可以(建议安装pytorch或者tensorflow对应的cuda版本)

2、cuda下载

cuda官网 https://developer.nvidia.com/cuda-toolkit-archive【以cuda11.6.0为例】

在这里插入图片描述

选择符合自己电脑的本地下载,点击 ‘ exe(local) ’ —> ‘ Download ’

在这里插入图片描述

3、cuda安装

下载完成后双击打开

在这里插入图片描述

选择一个空文件夹(临时文件,安装结束后会自动删除)—> 点击 ‘ OK ’

在这里插入图片描述
在这里插入图片描述

选择 ‘ 自定义安装 ’

在这里插入图片描述

取消勾选 “ Visual Studio Integration ”(除非你有C++环境);点击 “ 下一步 ”

在这里插入图片描述

选择要安装的位置,建议新建文件夹。点击 “ 下一步 ”

在这里插入图片描述

安装完成

在这里插入图片描述

4、cuDNN下载及安装

下载地址:https://developer.nvidia.com/rdp/cudnn-download

注册登录后下载对应cuda版本的cuDNN

如果是对应其他版本,选择 “ Archived cuDNN Releases ”

在这里插入图片描述

下载完成后解压

在这里插入图片描述

将文件夹下的内容复制到CUDA同名的文件夹下;不是替换!!!

注意:lib是win64下的内容
在这里插入图片描述

5、环境变量配置

搜索‘编辑环境变量’
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

6、检验是否配置成功

win+R键;输入 “ cmd ” ,点击 “ 确定 ”

在这里插入图片描述

输入“ nvcc -V ”

在这里插入图片描述

配置成功!

7、多CUDA

按照1-6的步骤下载所需CUDA及对应cuDNN

打开环境变量,新建后将想要用的CUDA版本置顶即可

在这里插入图片描述

win+R键;输入 “ cmd ” ,点击 “ 确定 ”;输入“ nvcc -V ”

在这里插入图片描述

显示11.2,配置成功!

三、Pytorch_gpu安装

1、Pytorch_gpu下载:https://pytorch.org/get-started/previous-versions

以CUDA11.6+pytorch_V1.12.0为例

在这里插入图片描述

不要直接在环境里输入命令;很容易下成CPU版本的

# CUDA 11.6 √√
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.6 -c pytorch -c conda-forge
  • 1
  • 2

直接下载whl包:https://download.pytorch.org/whl/torch_stable.html

ctrl+F键,输入cuda版本和你的python版本:例如:cu116-cp38

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

pip install +whl文件路径


# CUDA 11.6 √√
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.6 -c pytorch -c conda-forge
pip install "D:\E_files\torch_install\torch-1.12.0+cu116-cp38-cp38-win_amd64.whl"
pip install "D:\E_files\torch_install\torchaudio-0.12.0+cu116-cp38-cp38-win_amd64.whl"
pip install "D:\E_files\torch_install\torchvision-0.13.0+cu116-cp38-cp38-win_amd64.whl"
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6

在这里插入图片描述
在这里插入图片描述

四、安装tensorflow_gpu

不行,不是GPU版本
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow
  • 1
  • 2

win11显卡RTX4060用不了cuda10,下载cuda11.2的版本

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

不用安装keras已经有了

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/2023面试高手/article/detail/398064
推荐阅读
相关标签
  

闽ICP备14008679号