安装Pytorch即进入Pytorch官网,选择Pytorch版本以及CUDA版本,按照給出的命令在PyCharm的Terminal输入命令安装,前提是PyCharm以及配置了对应的anaconda环境。
pip3 install torch==1.10.2+cu102 torchvision==0.11.3+cu102 torchaudio===0.10.2+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.html
安装完成后,执行import torch查看Pytorch是否安装成功,执行torch.cuda.is_available(),若结果为True说明CUDA也配置无误。
同时发现PyCharm里的Python Console类似于Python自带的编译器,可用来测试一些短小的指令或者对陌生的函数进行试运行,右侧还会出现当前环境变量的值,类似于调试,很方便。
接下来学习Jupyter的使用,Jupyter是交互性的Python编辑器,可以将程序分段执行,方便更改与调试程序,很方便。
首先在刚刚安装Pytorch的虚拟环境中安装Jupyter,在PyCharm的Terminal中执行命令conda install nb_conda,安装完毕后,使用jupyter notebook指令,点击链接进入Jupyter环境,选择安装了Pytorch的虚拟环境,创建新的文件,可以开始编程了。