教程—如何快速安装最新版本的深度学习库Pytorch 2.3.0

由于网上的教程鱼龙混杂且大多是前几年的教程,本文主要介绍如何在Window系统中安装2024年最新版本的深度学习库Pytorch 2.3.0。

(1)Pycharm和Anaconda软件安装。

这里假设已经安装好Pycharm和Anaconda软件。我的Pycharm版本是PyCharm Community Edition 2021.3.1;Anaconda版本是Anaconda3-2022.10-Windows-x86_64版本

(2)然后需要创建新环境。

打开Anaconda prompt,输入下面的代码,自动在下面的目录新建名为pytorch230的环境,python版本选择3.10。

conda create -p E:\ProgramFlies\Anaconda3\envs\pytorch230 python=3.10

创建成功后,打开Anaconda软件,发现多了个名为pytorch230的环境,说明环境创建成功;或者通过下面的代码查看已经安装的环境:

conda env list

(3)在环境中安装pytorch。

记得科学上网(不然后面下载网速很慢),然后打开PyTorch官网。

选择下面的,选择对应的版本和CUDA版本(我的电脑CUDA是12.1版本,查看方法可以看其他教程):

打开Anaconda prompt,输入下面的代码,等待下载(文件大小接近2G),估计5分钟左右,即可完成自动安装:

conda activate pytorch230

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

然后安装其他常用的库到pytorch230环境中:

conda install matplotlib==3.5.1

conda install pandas

conda install scipy

conda install imageio

当然安装完要检查能否调用GPU:测试如下,如果输出TRUE说明pytorch能正确调用GPU,说明安装成功。

conda activate pytorch230

python

import torch

print(torch.__version__)

print(torch.cuda.is_available())

(4)最后,关联Pycharm和已经创建的pytorch230环境。

在Pycharm中,新建一个python项目,把环境关联到下面这个项目中(如图)。就可以愉快地在Pycharm写代码(bug)了。


 

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值