参考https://blog.csdn.net/qq_23013309/article/details/103965619
安装Pytorch
官网 https://download.pytorch.org/whl/torch_stable.html 选择合适的版本
torch/torchvision 都需要安装
下载了安装包,在软件安装包/GPU-banben
conda create -n pytorch python=3.7
conda activate pytorch
如果不想进入下载目录,可以直接利用以下方法。
注:pip install命令后面接的路径,都可以通过鼠标右键文件名-属性-安全-对象名称获取
pip install D:\软件安装包\gpu-banben\torch-1.2.0-cp37-cp37m-win_amd64.whll
pip install D:\软件安装包\gpu-banben\torchvision-1.2.0-cp37-cp37m-win_amd64.whll
验证成功安装,注意version是双下划线。
同理在服务器上安装Linux版本的pytorch成功。