当前位置:   article > 正文

Windows环境下Pytorch环境搭建_windows配置pytorch环境

windows配置pytorch环境

一、下载安装Miniconda

镜像文件地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/

2.将Miniconda相关路径添加至系统变量的路径中。

  

3.打开Anaconda Powershell Prompt,输入:conda --version

   

二、新建Python3.9虚拟环境

1.新建虚拟环境命令: conda create -n conda_name python=x.x(带python版本的)

新建python39环境: conda create -n python39 python=3.9

进入python39环境: conda activate python39

2.添加清华镜像网站到Anaconda:

conda config --add channels http://mirror.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

conda config --add channels http://mirror.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config --set show_channel_urls yes

3.如果需要 jupyter notebook可以使用conda或者pip命令进行安装:

pip install jupyter notebook

4.在当前目录下输入 jupyter notebook即可打开。

三、下载英伟达驱动

1. 下载nvidia驱动软件:官网下载,登录,下载驱动。(判断自己的电脑是否有英伟达的显卡,没有显卡不需要安装驱动。) https://www..cn/geforce/drivers/

windows+R,输入cmd,打开命令行,输入nvidia-smi,查看显卡驱动信息。

四、安装CUDA版Pytorch

1.判断自己的电脑是否有英伟达的显卡,没有显卡不需要安装CUDA。

确定显卡算力 显卡由 GPU、显存等组成,大部分情况下我们所说的 GPU 就等同于显卡,但是实际情况是 GPU 是显示卡的“心脏”、核心零部件、核心组成部分。

GPU 本身并不能单独工作,只有配合上附属电路和接口才能工作,这时候,它就变成了显卡;

显卡算力是指显卡处理信息的能力。

 确定显卡算力 找到显卡对应算力:https://developer.nvidia.com/cuda-gpus

2.在pytorch官网选择你要安装的版本信息,选择pip或者conda安装方式,然后下面会显示你需要安装的安装命令。 官网地址:https://pytorch.org/get-started/locally/

复制代码到powershell,注意要删除-c pytorch(不删除默认使用官网网站地址下载,删除之后下载要快一些,网速好也可以不删除):

#原本 conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

#删除之后 conda install pytorch torchvision torchaudio pytorch-cuda=12.1  -c nvidia

查看是否安装成功:import torch print(torch.__version__)

五、CPU版本pytorch安装

1. 在pytorch官网选择你要安装的版本信息,选择pip或者conda安装方式,选择CPU版本,然后下面会显示你需要安装的安装命令。

2.复制代码到powershell,注意要删除-c pytorch(不删除默认使用官网网站地址下载,删除之后下载要快一些,网速好也可以不删除):

#原本 conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

#删除之后 conda install pytorch torchvision torchaudio pytorch-cuda=12.1  -c nvidia

查看是否安装成功:import torch print(torch.__version__)

六、下载并配置Pycharm

1.进入pycharm官网,下载最新版pycharm(专业版自己网上找激活教程) 官网地址:https://www.jetbrains.com/pycharm/download/?section=windows

2.打开pycharm,点击设置:

   

点击 conda environment (conda 环境)

选择conda可执行文件:conda安装目录下\condabin\conda.bat,点击 load environements (加载环境)

选择 Using existing environment(使用现有环境),选择自己的虚拟环境。

安装完毕:

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

闽ICP备14008679号