【PYTORCH,TENSORFLOW环境配置,安装,自用代码】

conda -V(查看版本,这步不要也罢)

conda create -n test python=3.7(创建环境)

conda activate test(激活)

conda env list(查看自己的环境)

nvidia-smi(查看是否可以安装GPU版本)

PyTorch(看自己能安装什么版本)

复制代码
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121(本人用的)

pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple

(之前的,加了清华大学的镜像源)

Tensorflow安装

conda search -f cudatoolkit

conda search -c conda-forge cudatoolkit

conda install cudatoolkit==11.0.3 -c conda-forge

conda search -c conda-forge cudnn

conda install cudnn=8.0.5.39 -c conda-forge(本人用的,还可以用更新的)

在 Windows 环境中从源代码构建 | TensorFlow (google.cn)

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

(真麻烦啊)

PS:用终端安装才是真快

!!!!!安装真的好麻烦,好麻烦!!!!!!!!!!!!!!!!!!!!!!!!!!谁懂,并不是为了省淘宝安装费,而是配置环境是最基础的!!!!!!(不得不学会)

(按照顺序无脑打得了,已经写的足够详细了,腰疼,配置了好几天,因为一些很愚蠢的问题,所以这些是精华)

相关推荐
番石榴AI6 分钟前
基于机器学习优化的主图选择方法(酒店,景点,餐厅等APP上的主图展示推荐)
图像处理·人工智能·python·机器学习
国产化创客18 分钟前
基于AI大模型智能硬件--小智AI项目PC端部署测试
人工智能
海边夕阳200619 分钟前
【每天一个AI小知识】:什么是零样本学习?
人工智能·经验分享·学习
平凡而伟大(心之所向)23 分钟前
云架构设计与实践:从基础到未来趋势
人工智能·阿里云·系统架构·安全架构
数据与后端架构提升之路24 分钟前
构建一个可进化的自动驾驶数据管道:规则引擎与异常检测的集成
人工智能·机器学习·自动驾驶
qq74223498432 分钟前
Python操作数据库之pyodbc
开发语言·数据库·python
xier_ran43 分钟前
Transformer:Decoder 中,Cross-Attention 所用的 K(Key)和 V(Value)矩阵,是如何从 Encoder 得到的
深度学习·矩阵·transformer
2401_841495641 小时前
【自然语言处理】轻量版生成式语言模型GPT
人工智能·python·gpt·深度学习·语言模型·自然语言处理·transformer
梵得儿SHI1 小时前
(第三篇)Spring AI 基础入门:PromptTemplate 与对话工程实战(从字符串拼接到底层模板引擎的进阶之路)
人工智能·prompt·大模型应用·spring ai·prompttemplate·ai 响应的质量与准确性·上下文管理策略