由于网络问题,直接采用conda install命令可能无法直接下载对应的cuda包。
方法一:采用pip命令替代
步骤1:切换pip的源为国内源:
若只是临时切换:
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple some-package
若永久切换pip源:
pip install pip -U
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
若pip 默认源的网络连接较差,可以临时使用镜像站来升级 pip:
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pip -U
临时切换和永久切换二选一即可。
步骤二:去pytorch官网下载
去官网选择对应的命令和cuda版本下载。