从0开始安装pytorch和TensorFlow环境

学校的环境已经搭载好了,因为开学延期,可能要5月初才能开学,这几天想在自己的PC上玩玩卷积神经网络,忘记当时怎么搭载的pytorch和TensorFlow环境,导致走了不少弯路。以下基于win10 纯净版。

1. 安装anaconda3,以管理员身份运行。添加环境,在安装的时候要打勾。


2. 安装Nvidia 驱动,和control panel英伟达控制面板,查找cuda版本


3. 打开anaconda prompt,Anaconda-Pytorch中Python3.7创建环境安装  conda create –n pytorch python=3.7

4. 添加清华镜像源

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/

(这步一定放到最后,报错原因不详)

最后执行pytorch安装

conda install pytorch torchvision cudatoolkit=10.1 (pytorch官网查找cuda版本对应型号,上图我的cuda是10.1)

5. 同样方法构建TensorFlow环境,用pip安装

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==2.0.0-alpha0

6. 输入conda list查看包是否齐全

7. pip install --upgrade 要升级的包名

8. pip install future

9. windows下输入 pip --default-timeout=100 install tensorboard

linux下输入 pip --default-timeout=100 install -U  tensorboard

以上顺带解决终端未响应报错的问题

依然报错

pip install future

10. 安装scikit-image

Pip install scikit-image -i https://pypi.tuna.tsinghua.edu.cn/simple

11.安装tqdm

Pip install tqdm -i https://pypi.tuna.tsinghua.edu.cn/simple

至此,基本上已经安装了pytorch框架所需的大部分包

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容