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

相关推荐
LeeZhao@5 分钟前
【狂飙全模态】狂飙AGI-Wan2.1文生视频实战部署-Gradio篇
人工智能·语言模型·音视频·agi
roman_日积跬步-终至千里6 分钟前
【人工智能导论】07-学习-CNN:卷积+池化+多层堆叠,有效处理图像等结构化数据
人工智能·学习·cnn
Coder_Boy_6 分钟前
【人工智能应用技术】-基础实战-小程序应用(基于springAI+百度语音技术)智能语音控制-单片机交互
人工智能·单片机·小程序
刘海东刘海东6 分钟前
结构学习理论和结构推理理论(提纲)
人工智能
不被AI替代的BOT18 分钟前
AgentScope深入分析-LLM&MCP
人工智能·后端
Jorunk20 分钟前
状态对齐是连接 GMM-HMM 和 DNN-HMM 的核心桥梁
人工智能·神经网络·dnn
袋鼠云数栈29 分钟前
媒体专访丨袋鼠云 CEO 宁海元:Agent元年之后,产业需回到“数据+智能”的长期结构
大数据·人工智能
Wishell201536 分钟前
日拱一卒之Python与matlab的内存读取区别
pytorch
TF男孩38 分钟前
一堆3D点,神经网络是怎么判断它是椅子的?
人工智能·神经网络
AI即插即用42 分钟前
即插即用系列 | CVPR 2024 RMT:既要全局感受野,又要 CNN 的局部性?一种拥有显式空间先验的线性 Transformer
人工智能·深度学习·神经网络·目标检测·计算机视觉·cnn·transformer