Linux系统下的pytorch环境配置

一、安装conda

安装conda的方法请参考博主的另一篇文章，里面详细的讲解了安装的方法。文章地址如下：

在Linux系统配置yolov5环境（tensorflow-gpu、cuda、cudnn）_安瑜丶的博客-CSDN博客

二、配置pytorch环境

1.创建虚拟环境

conda create –n pytorch python=3.8

2.进入虚拟环境，安装pytorch

conda activate pytorch

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch

3.验证是否安装好

import torch
import torchvision
print(torch.cuda.is_available())
a = torch.Tensor(5, 3)
a = a.cuda()
print(a)
print(torch.__version__)

结果如图所示，即已成功安装。

4.安装需要的相关库

所需要的库如下：

scipy==1.7.1
numpy==1.21.2
matplotlib==3.4.3
opencv_python==4.5.3.56
torch==1.7.1
torchvision==0.8.2
tqdm==4.62.2
Pillow==8.3.2
h5py==2.10.0
tensorboard
scikit-learn
Cython
如果想更快速的安装可以在一个文件夹里创建一个requirements.txt文件，复制上述内容到txt文件中。并执行如下命令，便可以一起安装(xx为文件路径)：

pip install -r xx/requirements.txt

三、总结

以上就是博主自己安装的方法了，希望对大家有所帮助，如果有什么问题，请在评论区留言，博主会积极为大家解答。