安装步骤
我的ubunntu系统(20.04.2)下安装有Miniconda
在pytorch官网
版本选择稳定版(stable)即可
我在ubuntu系统安装,故OS选择Linux
安装方式选择conda
语言选择python
我电脑没有装cuda驱动,所以选择CPU
将Run this command中的内容复制到ubuntu终端中执行即可。
花絮
安装方式我一开始选择的是pip,但是第一次由于网速太慢下载失败了,第二次出现了两个警告
其中一个警告为
WARNING: The scripts convert-caffe2-to-onnx and convert-onnx-to-caffe2 are installed in '/xxxx/python3/bin' which is not on PATH.
Consider adding this directory to PATH or, if you prefer to suppress this warning, use --no-warn-script-location.
另一个警告为
WARNING: The scripts f2py, f2py3 and f2py3.8 are installed in '/home/cyd/.local/bin' which is not on PATH.
Consider adding this directory to PATH or, if you prefer to suppress this warning, use --no-warn-script-location.
百度的解决方法是 gedit ~/.bashrc,export路径,source生效更改。尝试无果。
然后我试图卸载重装,发现之前好像根本没安装上?????
然后我就去pytorch官网选择conda方式安装,成功了,import也没错
但奇怪的是,conda env list没有显示pytorch,我也不知道为什么。
为了以防万一,我又用了查看pytorch版本的命令,是正常的。
import torch
print(torch.__version__)
说明
可以看到我所有命令都是在base环境下操作的,所以pytorch被安装到base环境下,使用conda ens list就找不到pytorch
使用下面这个命令就可以在minconda的env下看到pytorch了
conda create -n pytorch
但是这个时候pytorch下面还什么都没有。
如果不介意在base环境下的话就不需要再进行什么操作了。
如果需要pytorch环境(盖一个新房子)的话,在(pytorch)下重新安装一遍就行了
上图来自B站up主:我是土堆(如有侵权,请联系我删除)