conda create --name machinelearning python=3.7
activate machinelearning
进入官网 pytorch官网
复制命令
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch