ubuntu系统下,用conda或者pip方式安装pytorch的时候会需要如下指令:
pip install torch torchvision
conda的:
conda install pytorch torchvision cudatoolkit= cuda对应版本号 -c pytorch
conda安装的时候经常会因为源网址的问题出现源地址没有对应的资源
pip安装的时候速度非常慢,而且经常下的过程中就error了
可以使用下面的指令,替换一下临时的源地址下载,亲测稳定可行,速度也挺快。
pip install torch torchvision -i https://pypi.mirrors.ustc.edu.cn/simple/