注意替换清华源,否则直接安装速度会很慢。
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
cuda版本根据自己情况修改,我的是10.1。
conda install pytorch torchvision cudatoolkit=10.1
---------------------
版权声明:本文为CSDN博主「玻璃渣里嗑糖」的原创文章,遵循CC 4.0 by-sa版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/pw1623/article/details/90257347