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

相关推荐
kkai人工智能1 小时前
AI写作:从“废话”到“爆款”
开发语言·人工智能·ai·ai写作
づ安眠丶乐灬5 小时前
计算机视觉中的多视图几何 - 1
人工智能·vscode·计算机视觉
2503_928411565 小时前
项目中的一些问题(补充)
人工智能·python·tensorflow
MarkHD5 小时前
智能体在车联网中的应用 第1天 车联网完全导论:从核心定义到架构全景,构建你的知识坐标系
人工智能·架构
中科米堆6 小时前
塑料制品企业部署自动化三维扫描仪设备,解决注塑件变形问题-中科米堆CASAIM
人工智能
星图云6 小时前
从数据累积到精准解析:AI解译打造遥感数据高效利用新范式
人工智能·卫星遥感
飞哥数智坊6 小时前
AI 大厂的“护城河”,也会成为它们的束缚
人工智能·创业
BB_CC_DD7 小时前
超简单搭建AI去水印和图像修复算法lama-cleaner二
人工智能·深度学习
珠海西格电力7 小时前
零碳园区物流园区架构协同方案
人工智能·物联网·架构·能源