赞
踩
本文有用的前提:你是用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
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。