说明
其实这个很简单,为啥还记录一下呢,主要也是遇到一些坑,用于以后或者其他朋友避免走这些坑吧
在这之前,已经安装cuda10.2和cudnn7.6.5以及Anconada,同时进入默认是base环境下

1.创建虚拟环境

conda create -n pytorch_env python=3.7.0

如图所示:
ubuntu下使用conda安装pytorch最新版本_linux

2.激活虚拟环境

conda activate pytorch_env

输入以上命令之后,终端命令行前会出现(pytorch_env)字标,说明已经进入此环境下

3.安装pytorch

pip install --upgrade torch torchvision

最后安装成功

4.验证是否安装成功以及版本号

python
import torch
print(torch.__version__)

输出版本号:
1.6.0
说明安装成功