【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:用终端安装才是真快

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

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

相关推荐
头发够用的程序员13 小时前
从滑动窗口到矩阵运算:img2col算法基本原理
人工智能·算法·yolo·性能优化·矩阵·边缘计算·jetson
ydmy13 小时前
transformer超参数配置(个人理解)
人工智能·深度学习
武帝为此13 小时前
【数据清洗缺失值处理】
python·算法·数学建模
AI原来如此13 小时前
AI 编程助手常见问题 10 问 10 答
人工智能·ai·大模型·编程
zhangchaoxies14 小时前
如何在 Go 中安全复制接口指针所指向的值
jvm·数据库·python
曲幽14 小时前
FastAPI + Pydantic 模型终极实战手册:从能跑就行到固若金汤,这些技巧你一定用得上
python·fastapi·web·model·field·pydantic·validator·basemodel
计算机软件程序设计15 小时前
Python Flask工程目录解读
python·flask·工程目录解读
Ares-Wang15 小时前
Flask》》 Flask-OpenID 认证、 OpenID Connect (OIDC)
后端·python·flask
科技林总15 小时前
【论文】MemGPT: Towards LLMs as Operating Systems
人工智能
黎阳之光15 小时前
黎阳之光透明管理:视频孪生重构智慧仓储新范式
人工智能·算法·安全·重构·数字孪生