位置: IT常识 - 正文
推荐整理分享在anaconda下安装pytorch + python3.8+GPU/CPU版本 详细教程(在anaconda下安装python),希望有所帮助,仅作参考,欢迎阅读内容。
文章相关热门搜索词:在anaconda下安装tensorflow gpu,在anaconda下安装tensorflow gpu,在anaconda下安装opencv,在anaconda下安装opencv,在anaconda下安装python,在anaconda下安装opencv,在anaconda下安装python,在anaconda下安装python,内容如对您有帮助,希望把文章链接给更多的朋友!
没安装Anaconda的同学可以参考以下安装链接: https://blog.csdn.net/qq_45281807/article/details/112442577
按照安装CPU版本和GPU两个版本进行分类,一般运行程序建议使用CPU版本的,安装更方便。
二、安装CPU版本的Pytorch1.打开Anaconda Prompt,如下图所示:2. 使用清华镜像源网站注意!如果切换镜像后当出现下载不了的情况,就先切换默认源,然后再修改另一个可以使用的conda源(一定要先恢复默认,再换另一个!!!),所以我们先切回默认源:
conda config --remove-key channels(1)切换到国内的镜像源(不然下载速度可能太慢) 分别输入以下4行代码:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/conda config --set show_channel_urls yesconda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/配置好之后以后若需要使用清华镜像,无需再配置。
3.创建Pytorch环境创建环境pytorch,使用Python版本是3.8,其中pytorch是虚拟环境的名字,可以改变:
conda create -n pytorch_38 python=3.8之后,在加载过程中会弹出提示,输入 y,即可安装。 查看环境是否安装成功
conda info --envs可以看到包含四个环境,其中pytorch是刚刚创建的环境。
4.激活刚刚创建的pytorch环境activate pytorch_38激活刚刚创建的pytorch环境。
5.安装Pytorch根据自己的安装版本,在Pytorch官网寻找安装命令代码: Pytorch官网:https://pytorch.org/
如图所示版本则是复制:
conda install pytorch torchvision torchaudio cpuonly -c pytorch将复制的代码粘贴到命令行格式下,弹出提示,输入 y,即可完成安装,显示“done”。
6.测试是否安装成功输入python, 进入python
python输入import torch,测试pytorch是否安装成功,无报错则表示安装成功。
三、安装GPU版本(电脑有显卡)1.查看是否安装CUDA软件驱动(1)使用快捷键WIN+R,输入cmd (2)输入命令,nvidia-smi (3)如果显示CUDA的版本号则可以开始进行第3步操作,繁殖反之,参考下文的第二章,配置cuda驱动软件:
https://blog.csdn.net/qq_45281807/article/details/121294644
2.打开Anaconda Prompt,如下图所示:3. 使用清华镜像源网站注意!如果切换镜像后当出现下载不了的情况,就先切换默认源,然后再修改另一个可以使用的conda源(一定要先恢复默认,再换另一个!!!),所以我们先切回默认源:
conda config --remove-key channels(1)切换到国内的镜像源(不然下载速度可能太慢) 分别输入以下4行代码:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/pkgs/main/conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/pkgs/free/conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/cloud/conda-forge/conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/cloud/bioconda/配置好之后以后若需要使用清华镜像,无需再配置。
4.创建Pytorch环境创建环境pytorch,使用Python版本是3.8,其中pytorch是虚拟环境的名字,可以改变:
conda create -n pytorch_38 python=3.8之后,在加载过程中会弹出提示,输入 y,即可安装。 查看环境是否安装成功
conda info --envs可以看到包含四个环境,其中pytorch是刚刚创建的环境。
5.激活刚刚创建的pytorch环境activate pytorch_38激活刚刚创建的pytorch环境。
6.安装Pytorch根据自己的安装版本,在Pytorch官网寻找安装命令代码: Pytorch官网:https://pytorch.org/ 输入命令:
conda install pytorch torchvision torchaudio cudatoolkit=11然后等一下会显示是否继续进行,输入y,点回车,就可以安装啦!!
7. 查看CUDA是否可用:依次输入以下命令,如果输出结果为True,则安装成功。
pythonimport torchprint(torch.cuda.is_available())返回False解决办法 如果你按照以上步骤安装一般是没有问题的。 如果出现False,大家可以尝试以下步骤进行排除: 1、首先确定显卡是否支持CUDA,https://www.geforce.com/hardware/technology/cuda/supported-gpus 2、确保驱动升级到最新(确保驱动与cuda版本互相支持)
参考文章:
https://blog.csdn.net/qq_45281807/article/details/112442423 https://blog.csdn.net/qq_45281807/article/details/121294644
上一篇:vue3 + vite 性能优化 ( 从5s -> 0.5s )(vue3性能对比)
友情链接: 武汉网站建设