0
点赞
收藏
分享

微信扫一扫

P0-PyTorch环境的配置及安装

东林梁 2023-10-06 阅读 28

2023年9月27日

0、安装Anaconda

image-20230927182501120

出现(base)说明安装成功

1、有序的管理环境

不同项目可能对Python\Pytorch的版本要求不一样,为了能够很好地切换Python\Pytorch的版本,使用Anaconda集成的conda包,创建不同的环境、不同版本软件互相隔离。

conda create -n pytorch python=3.6
//-n表示名字,创建了一个名叫pytorch的环境,python=3.6表示房子里面的Python版本
conda activate pytorch
//激活此环境
pip list
//查看环境当中的有哪些工具包

image-20230927183350780

image-20230927185002292

发现工具包内并没有pytorch,需要安装pytorch官网->

https://pytorch.org/

image-20230927185655005

根据操作系统、GPU信号、下载方式(Linux通常采用pip,Windows通常采用Conda得到命令行。

为了正常使用CUDA,英伟达显卡驱动的版本号在396以上。

在cmd输入nvida-smi查看驱动版本号。

ps,CUDA(Compute Unified Device Architecture)是一种由NVIDIA开发的并行计算平台和应用程序编程接口(API)

image-20230927215558889

image-20230927220800889

pip list显示有torch说明安装成功,进入Python成功导入torch包显示成功,torch.cuda.is_available()检测是否可以正常使用GPU。

image-20230927221016202

torch.cuda.is_available()
举报

相关推荐

0 条评论