作为一个学习python的小白,为了安装一个pytorch包,浪费了一天的时间,一直在出现各种错误,一直在找各种解决办法,最后坚持到晚上十点半,终于解决!!记录一下过程,方便以后自己再次需要。
从一开始在prompt中参考文章使用以下命令创建了一个虚拟环境:
conda create -n torch python=3.9
代码中的torch表示自定义虚拟环境的名称,3.9表示的是本机安装的python版本,本人安装的是3.9.8版本,因此这里是3.9.进一步使用以下命令激活虚拟环境:
activate torch
虚拟环境成功激活以后可以看到最左边的单词由base变成了torch,表示虚拟环境激活成功。
激活环境以后根据自己电脑的配置在官方网站(https://pytorch.org/get-started/locally/)得到安装命令:
conda install pytorch torchvision torchaudio cpuonly -c pytorch
复制以上命令到 prompt中便可在设置的虚拟环境中安装pytorch,本人设备只支持CPU,于是在compute platform这里我选择的是CPU,这里可以根据自己的实际设备进行选择,最后输出done则表示安装成功。
安装成功在该虚拟环境下使用以下命令可以成功的导入torch包,但是在jupyter notebook里面导入则会报错no module named "torch".
import torch
问题就出在这里,我查阅了很多方法一直不知道该如何解决这个问题,反复重新安装pytorch,但仍然出现同样的问题,最后本人差点放弃,各种搜索解决办法,终于!!看见曙光,参考文章将设置的虚拟环境导入到jupyter内核中。即在以上操作的基础上先输入以下命令安装 ipykernel:
conda install -c anaconda ipykernel
然后使用以下命令将虚拟环境加到内核:
python -m ipykernel install --user --name=torch
以上命令完成以后便成功加入内核,就可以在jupyter中切换内核成功导入torch包了!!!