今天学习PCA的时候,找到了一个大佬的代码,用到了Pytorch,之前都是在服务器跑的,自己电脑没有装,所以今天就安装一下。
conda info --envs

conda create -n pytorch_env python=3.7


activate pytorch_env

https://pytorch.org/get-started/locally/

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch


注:此时点击运行可能会报错,“找不到模块”,然后发现并没有启动conda环境,也就是在命令行没有下面图片红框中的内容:

此时,可以按照这篇博客的方法解决:python之 vscode中激活conda虚拟环境报错或者打开虚拟环境失败。