其他分享
首页 > 其他分享> > DW组队学习——深入浅出PyTorch笔记

DW组队学习——深入浅出PyTorch笔记

作者:互联网

本篇是针对DataWhale组队学习项目——深入浅出PyTorch而整理的学习笔记。由于水平实在有限,不免产生谬误,欢迎读者多多批评指正。

安装PyTorch


安装Anaconda

这里为了避免手动安装Python及数据科学相关的众多的package,我们直接安装Anaconda,在此基础上再安装Pytorch。

官网直接下载安装即可,地址https://www.anaconda.com/products/distribution

创建虚拟环境

为了不与未来安装的其他版本的Python及package发生冲突,我们单独为PyTorch创建一个Anaconda中的虚拟环境。

win用户可以在Anaconda Prompt中利用指令进行创建和后续操作,详见DW教程https://datawhalechina.github.io/thorough-pytorch/index.html
在这里我直接利用图形界面的选项进行操作了,也很简单,点击Environments-Create,按要求进行创建即可

由于外网速度限制,我们通常给conda换第三方镜像源,比如清华源等。需要修改C:\Users\User_name.condarc文件,换成以下内容:

channels:
  - defaults
show_channel_urls: true
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

具体操作可以参考https://mirror.tuna.tsinghua.edu.cn/help/anaconda/
如果弄不成,也可以在每次conda install时指定来源

安装PyTorch

首先在cmd命令行中利用nvidia-smi查看自己显卡的型号,然后对应下表找到合适的CUDA版本

然后在PyTorch官网(https://pytorch.org/)上选择自己需要的版本,会自动生成一个conda指令,可以粘贴到新创建的虚拟环境的terminal中执行来完成安装

同样,虚拟环境可以用conda activate env_name指令激活并进入,也可以直接在图形界面中相应虚拟环境中open terminal

安装其他需要的package

同样在对应的虚拟环境中可以安装cuda及cudnn等包,注意版本要搞对

conda install cudatoolkit==版本号
conda install cudnn==版本号

(未完待续)

参考资料:

  1. https://datawhalechina.github.io/thorough-pytorch/index.html DataWhale——深入浅出PyTorch
  2. https://mirror.tuna.tsinghua.edu.cn/help/anaconda/ Anaconda镜像使用帮助

标签:cn,anaconda,组队,PyTorch,tsinghua,https,edu,DW,tuna
来源: https://www.cnblogs.com/liugd-2020/p/16593119.html