Windows下配置pytorch环境

最近跑模型时不小心安装了一些包,导致原来的依赖包都变了,运行时各种报错,最后采用最暴力解决办法:卸载Anaconda和pytorch,重新安装。写一篇博客记录安装anaconda和pytorch的方法。

Anaconda安装(推荐python3.7版本)

Anaconda官网:
最新下载地址(不推荐使用最新的python版本,因为最新版在安装一些依赖包时可能会出错)
Anaconda所有历史版本
将此版本下载到:
Windows下配置pytorch环境

pytorch安装

首先需要在anaconda的base环境中创建一个虚拟环境:

conda create -n pytorch python=3.7

Windows下配置pytorch环境
这里如果不清楚python版本号的话可以先在终端中输入python查看:
Windows下配置pytorch环境
创建完虚拟环境后,就可以上官网上下载pytroch了
Windows下配置pytorch环境

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

使用-c python 安装会强制进入python官网进行下载,速度非常慢。

清华园镜:

在终端中输入以下命令:

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 --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

最后运行:

conda install pytorch torchvision torchaudio cudatoolkit=10.2

最后,在pycharm上更换新创建pytorch环境:
Windows下配置pytorch环境
Windows下配置pytorch环境
Windows下配置pytorch环境

版权声明:本文为博主RooKiChen原创文章,版权归属原作者,如果侵权,请联系我们删除!

原文链接:https://blog.csdn.net/RooKichenn/article/details/123062643

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
青葱年少的头像青葱年少普通用户
上一篇 2022年2月23日 下午3:09
下一篇 2022年2月23日

相关推荐