深度学习用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,就安装好了。

相关推荐
QxQ么么5 分钟前
移远通信(桂林)26校招-助理AI算法工程师-面试纪录
人工智能·python·算法·面试
愤怒的可乐41 分钟前
从零构建大模型智能体:统一消息格式,快速接入大语言模型
人工智能·语言模型·自然语言处理
每天一个java小知识2 小时前
AI Agent
人工智能
猫头虎2 小时前
如何解决 pip install 编译报错 fatal error: hdf5.h: No such file or directory(h5py)问题
人工智能·python·pycharm·开源·beautifulsoup·ai编程·pip
龙赤子2 小时前
人工智能AI的大框架
人工智能
比奥利奥还傲.2 小时前
本地+AI+大模型自由用!Cherry+Studio打破局域网限制
人工智能
雪碧聊技术2 小时前
深度学习、机器学习、人工智能三者的关系
人工智能·深度学习·机器学习
β添砖java2 小时前
机器学习初级
人工智能·机器学习
陈奕昆2 小时前
n8n实战营Day3:电商订单全流程自动化·需求分析与流程拆解
大数据·开发语言·人工智能·自动化·需求分析·n8n
努力改掉拖延症的小白2 小时前
Intel笔记本也能部署大模型(利用Ultra系列gpu通过优化版ollama实现)
人工智能·ai·语言模型·大模型