一、安装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文章来源：https://www.toymoban.com/news/detail-614421.html
三、总结
以上就是博主自己安装的方法了，希望对大家有所帮助，如果有什么问题，请在评论区留言，博主会积极为大家解答。文章来源地址https://www.toymoban.com/news/detail-614421.html
到了这里，关于Linux系统下的pytorch环境配置的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！
