P0-PyTorch环境的配置及安装

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
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
1. 安装Anaconda 首先,我们需要安装Anaconda,它是一个Python的集成环境,可以方便地管理Python的包和环境。可以从Anaconda官网下载对应的安装包进行安装。 2. 创建虚拟环境 在Anaconda中,我们可以通过创建虚拟环境来管理不同的Python环境和包。打开Anaconda Prompt,输入以下命令创建一个名为pytorch的虚拟环境: conda create -n pytorch python=3.7 3. 激活虚拟环境 创建好虚拟环境后,我们需要激活它才能使用。在Anaconda Prompt中输入以下命令激活虚拟环境: conda activate pytorch 4. 安装PyTorch 在激活虚拟环境后,我们可以使用conda命令安装PyTorch。根据自己的需求选择对应的版本和安装方式。例如,如果需要安装CPU版本的PyTorch,可以输入以下命令: conda install pytorch-cpu torchvision-cpu -c pytorch 如果需要安装GPU版本的PyTorch,可以输入以下命令: conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge 5. 配置PyCharm 打开PyCharm,选择File -> Settings -> Project -> Project Interpreter,在Interpreter中选择刚才创建的虚拟环境pytorch。如果没有找到虚拟环境,可以点击右侧的Show All,然后选择对应的Python解释器。 6. 测试PyTorch 在PyCharm中创建一个Python文件,输入以下代码测试PyTorch是否安装成功: import torch print(torch.__version__) 如果输出了PyTorch的版本号,则说明安装成功。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值