深度学习用pytorch 使用gpu 创建anaconda环境

文章目录

创建环境

bash 复制代码
conda create -n 环境名称 python=3.8

激活新建的环境

bash 复制代码
conda activate 环境名称

安装需要的平时的库(不包括pytorch)

bash 复制代码
pip install numpy

自己根据需要安装

查看电脑的显卡驱动版本+安装cuda

在cmd命令窗口中输入nvidia-smi,查看CUDA Version

根据这个里面安装好CUDA 全网最详细的安装pytorch GPU方法,一次安装成功!!包括安装失败后的处理方法!

,终端输入nvcc -V 查看是否安装好。

安装pytorch

pytorch网站 根据自己电脑情况,选择pytorch。

复制pip语句到终端,进行安装。

测试pytorch是否用gpu

python 复制代码
import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.cuda.is_available())  #输出为True,则安装无误

终端输出为True,就安装好了。

相关推荐
智恒百亿6 分钟前
RTX 5090 深度解析:8 卡智算集群多卡协同痛点完整解决方案
人工智能
树獭非懒6 分钟前
七、智能体的Reflection机制:让AI学会三省吾身
人工智能·llm·agent
字节跳动视频云技术团队9 分钟前
不止于 4K,火山引擎画质增强让视频从清晰走向细腻
人工智能·音视频开发
中微极客18 分钟前
解锁LLM开发全栈能力:Python + LangChain + RAG 工程实战指南
人工智能·python·langchain
CCC:CarCrazeCurator22 分钟前
A100作为深度学习基础计算节点的原因
人工智能·深度学习
米小虾28 分钟前
操作系统正在被AI重写:从谷歌到微软,为什么所有人都在押注"Agent OS"?
人工智能
ACP广源盛1392462567335 分钟前
IX8024@ACP# 搭配此芯 AI 服务器 + 爱芯元智产品完整方案
大数据·运维·服务器·人工智能·分布式·嵌入式硬件
hhzz38 分钟前
Barbershop:基于GAN和分割Mask的图像合成技术——从理论到实战全解析
图像处理·人工智能·python·深度学习·计算机视觉
qyr67891 小时前
全球新能源汽车电机驱动器市场深度调研报告
大数据·网络·人工智能·自动化·汽车
动物园猫2 小时前
校园异常行为目标检测数据集:5类别 | 目标检测
人工智能·目标检测·计算机视觉