由于网上的教程鱼龙混杂且大多是前几年的教程,本文主要介绍如何在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)了。