安装pytorch-gpu踩坑

最新推荐文章于 2024-08-11 00:03:20 发布

攻城狮要飞翔

最新推荐文章于 2024-08-11 00:03:20 发布

阅读量287

点赞数

分类专栏：深度学习_图像处理_语义分割文章标签：神经网络深度学习 pytorch

本文链接：https://blog.csdn.net/lovebabyhai/article/details/107689246

版权

深度学习_图像处理_语义分割专栏收录该内容

4 篇文章 1 订阅

订阅专栏

据说中科院镜像已经不能用了，可以用清华镜像。

遇到问题：

加载清华镜像：

还是不行，可能是因为配置了乱七八糟的url，重置并添加清华镜像

# 重置源配置
conda config --remove-key channels

# 重新添加清华源
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/pytorch/ 
conda config --set show_channel_urls yes

随后下载pytorch：

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch