anaconda3+PyCharm下pytorch环境快速搭建_whl文件下载
当运行一个新的网络的时候,往往需要不同的环境,本文结合多篇博客总结了个人认为最为简单快速的环境搭建方法,通过whl文件下载避免了因网络差而导致的安装失败。
前提:
1你已经根据自身电脑硬件条件,下载好 anaconda3和PyCharm
2 建议提前安装cudnn、cuda及其驱动
1 开始菜单中打开Anaconda3下的Anaconda Prompt
2 新建虚拟环境命令
conda create -n your_env_name python=X.X
your_env_name:你的环境名称
python==X.X:你需要的Python版本号
3 下载好torch和torchvision对应的whl
个人觉得以下网址十分实用,力荐收藏:https://download.pytorch.org/whl/torch_stable.html
- 注意:cu开头的为GPU版本,cpu开头的是CPU版本
4 whl文件保存
在anaconda文件夹下找到envs文件夹,会看到自己新建的环境文件夹your_env_name,在your_env_name下新建文件夹,比如RunTorch,将torch和torchvision的whl文件存在里面,便于后续安装管理。
5 在新建环境下安装pytorch
- 激活环境
activate your_env_name
- 进入安装目录(比如环境在E盘)
- 切换盘
E:
- 进入第4步保存的文件夹下
cd E:/anaconda/envs/your_env_name/RunTorch
- 切换盘
- 输入下载命令
pip install A.whl
A是RunTorch文件夹下保存的文件名,注意在这里我们先下载torch再下载torchvision
6 在新环境下安装需要的包
pip install 模块名==版本号
7 退出环境
deactivate your_env_name
exit()
一个新的环境就搭建好啦,在pycharm中通过Add Interpreter加载新建的环境,加载过后打开新的工程时就可以直接切换啦!