pytorch之旅

一、环境准备

先准备安装Anaconda，环境用的3.7版本的。创建pytorch环境之后启用pytorch环境

conda create -n pytorch python=3.7
conda activate pytorch

然后看下用哪些包：

conda activate pytorch

下一步，安装pytorch。

登录官网，https://pytorch.org/，配置界面：

这里需要查看自己gpu的型号，通过任务管理器就可以看到了：

 输入这段代码看下显卡的版本号：

nvidia-smi

下面会有生成的一段命令：

conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia

如果想用回原来的源：

conda config --remove-key channels

不得不说装个cuda还挺麻烦，报错，算了先用cpu版，技术娴熟了在用cuda版：

conda install pytorch torchvision torchaudio cpuonly -c pytorch

查看安装包情况：

pip list

如果安装包比较慢，也有个办法，就是直接放到pkg里面，我是安装到D盘的：

D:\Program Files\Anaconda3\pkgs

还在下在gpu的包，没有gpu的时候，测试这段代码就会为false：

(pytorch) C:***>python
Python 3.9.16 (main, Jan 11 2023, 16:16:36) [MSC v.1916 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
False

对于pycharn，设置和anaconda一样的环境，有个办法，用env/pytorch下面的python就可以保持环境的一致性。

这里也补充下安装gpu版本的pytorch，把版本号删了，随他下！

conda install pytorch torchvision torchaudio -c pytorch -c nvidia

然并卵，还是没成功，后续边研究试试。

然后是在pytorch环境中安装jupyter notebook：

(pytorch) C:\Users\zz>conda install nb_conda

安装完之后便可以启动jupyter notebook

jupyter notebook