【深度学习】PyTorch环境搭建(Windows)
PyTorch环境搭建(Windows)
·
一、给conda添加清华源优化下载
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#下载时显示文件来源
二、在官网获取conda命令行进行下载
在Pytorch官网选择对应版本,复制对应语句,在Anaconda对应环境下执行。(CUDA版本可用nvcc -V命令查询)
注意,语句需去除-c及后面部分,若不删掉,则还是在default源下载。
以本人适合的环境为例:conda install pytorch torchvision torchaudio cudatoolkit=11.3
三、下载成功!

四、测试
版本号成功输出,环境配置成功!
附录 - Jupyter Notebook中输出版本号报错问题
在Anaconda Navigator中选择对应环境,下载Jupyter Notebook。
下载完成后,在开始菜单点击对应Jupyter Notebook启动。
即可成功!
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐

所有评论(0)