开启清华源。重要的是https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ 这个源一定要添加进去,然后才能正常安装pytorch
这是conda源,国内好像也就只有清华这么一家conda源.
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
使用命令 nvcc --version 可以查看cuda的版本
然后到这里去生成安装需要的命令 https://pytorch.org/get-started/locally/
安装GPU版的如下. 我的cuda版本是10.2所以我的命令是下面的
conda install pytorch torchvision cudatoolkit=10.2 # -c pytorch 这个要去掉
然后执行命令 注意删掉最后的参数 -c pytorch
如果要安装无GPU的版本的命令如下
conda

最低0.47元/天 解锁文章
1万+

被折叠的 条评论
为什么被折叠?



