cuda90、cuda10 安装pytorch1.4
第一步:anaconda创建虚拟环境
conda create -n env_name python=3.6
conda activate env_name
第二步: 切换pytorch的清华源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
第三步: 安装对应版本pytorch
conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.0
第四步:查看是否安装成功
#cmd输入 python
import torch
print(torch.__version__)
大功告成
大家加油!!