0
点赞
收藏
分享

微信扫一扫

Pytorch学习(基本环境配置)(一)

Hyggelook 2022-01-10 阅读 50

下载anaconda

官网链接:https://www.anaconda.com/
Anaconda个版本:https://repo.anaconda.com/

安装最新版即可,安装的后面跳过安装【Install Microsoft VsCode】。

显卡

打开任务管理器,点击【性能】->【GPU】,如果能正确显示型号则说明显卡安装成功。

nvidia-smi #查看显卡相关信息

anaconda环境管理

conda create -n pytorch python=3.8  #创建名字为pytorch的环境
conda activate pytorch   #激活(使用)该环境
conda deactivatte     #不使用(激活)该环境
pip list   #看环境中有哪些包

pytorch安装

pytorch下载:https://pytorch.org/
按照默认下载即可
复制指令:conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch在anaconda prompt里运行即可。

在 anaconda prompt中运行:

python
import torch
torch.cuda.is_available()

不报错说明pytorch安装成功,gpu可以使用。

pycharm

pycharm下载链接:https://www.jetbrains.com/pycharm/默认安装即可。

pycharm选择已有的conda环境:
【File】【Settings】【Project:(项目名字)】【Python Interpreter】【点击后面的齿轮(设置)按钮】【Add】【Conda Environment】【Existing environment】
选择现有的解释器:anaconda创建的解释器位置在C:\Users\user\Anaconda3\envs

在pycharm的Python Console中输入:

import torch
torch.cuda.is_available()

输出True就没问题。Python Console可以像matlab一样看到所有变量的信息。

jupyter notebook

打开jupyternotebook,在anoconda prompt中输入jupyter notebook e:即可以e盘为主目录打开。

jupyter 使用conda环境

在anaconda prompt中激活要使用的环境,再安装nb_conda这个包。

conda activate pytorch
conda install nb_conda

再在jupyter notebook中选择pytorch这个环境就好了

举报

相关推荐

0 条评论