从零搭建torch环境,保姆级教程,新手小白,进进进!!!

1.管理创建环境,name后面的名字,自己定义,这里wxy为例

conda create --name wxy python=3.7.10

2.激活并进入环境:

avtivate wxy

3.xxx库的安装:进入环境后,输入:pip install xxx -i Simple Index,后面一群代码为清华镜像源,加速安装。

观察自己代码需要torch还是tensorflow,注意安装torch或者tensorflow时,要安装与CUDA相对于对应的版本号。

以cuda11.1版本为例

安装tensorflow:pip install tensorflow-gpu==2.4.1 -i https://pypi.tuna.tsinghua.edu.cn/simple

安装torch:pip install torch==1.8.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

4. pip list 可以查看自己的环境里安装了哪些东西。

如何进行实战,请参考下一篇博文:搭建好自己的网络后,如何搭建一个完整的网络进行实战训练

猜你喜欢

转载自blog.csdn.net/weixin_61745097/article/details/125272686