pytoch 安装

注意替换清华源,否则直接安装速度会很慢。

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

原文地址:https://www.cnblogs.com/find1/p/11327786.html