当前位置:   article > 正文

CUDA的安装 — 用自己电脑的GPU跑深度学习模型_显卡跑深度学习

显卡跑深度学习

本文有用的前提:你是用Anaconda装的python开发环境,用pytorch构建模型。

如何用Anaconda装的python开发环境,请参考另一篇博客,链接如下

 Anaconda + Pycharm,利用Anaconda安装python并配置虚拟环境,包括sklearn、pytorch的安装_learninger_lt7的博客-CSDN博客

一、Win + R, cmd —> 输入

nvidia-smi

找到

这里的CUDA Version指的是你的电脑最高支持的CUDA版本,它是向下兼容的,因此,低于12.0版本的CUDA都可安装。

二、进入pytorch官网,PyTorch

往下滑,找到这个,按我这么选

建议选择CUDA11.7版本,我试过11.8,莫名的报错。 

 复制灰色部分的代码。

三、打开开始菜单,找到Anaconda Powershell Prompt,点击进入

 输入:conda activate myEnvs_3_7_16  # (红色部分应根据你自己的虚拟环境名填写)

成功激活虚拟环境后,接着在红线处粘贴刚刚复制的那段代码,然后回车等待安装即可。

如果提示pip版本问题,就在当前界面(虚拟环境已激活的情况下)升级pip,升级pip的代码如下:

python -m pip install --upgrade pip

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

闽ICP备14008679号