说明
其实这个很简单,为啥还记录一下呢,主要也是遇到一些坑,用于以后或者其他朋友避免走这些坑吧
在这之前,已经安装cuda10.2和cudnn7.6.5以及Anconada,同时进入默认是base环境下
1.创建虚拟环境
conda create -n pytorch_env python=3.7.0
如图所示:
2.激活虚拟环境
conda activate pytorch_env
输入以上命令之后,终端命令行前会出现(pytorch_env)字标,说明已经进入此环境下
3.安装pytorch
pip install --upgrade torch torchvision
最后安装成功
4.验证是否安装成功以及版本号
python
import torch
print(torch.__version__)
输出版本号:
1.6.0
说明安装成功