PyTorch学习

Posted 2021-02-28 pengweiblog

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了PyTorch学习相关的知识，希望对你有一定的参考价值。

安装

新建环境：

conda create --name torch python=3.7

安装一些可能要用到的包（非必须，看自己情况）：

conda install numpy
conda install spyder
conda install jupyter notebook

安装PyTorch：

conda install pytorch torchvision cpuonly -c pytorch # CPU版

GPU版根据CUDA版本不同命令也不同，可以去这里查看安装命令

以上是关于PyTorch学习的主要内容，如果未能解决你的问题，请参考以下文章