pytorch

1.安装CUDA

首先确认你的电脑是否有GPU

终端输入

1

nvidia-smi

查看驱动信息,查看叫CUDA Version的

进官网下载CUDA

下载地址：https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&Distribution=WSL-Ubuntu&target_version=2.0&target_type=deb_local

你也可以在终端下载(此安装包4个G)

1

sudo apt install nvidia-cuda-toolkit

下载完之后输入

1

nvcc -V

查看CUDA版本（我这边是11.5）

接下来去到pytorch官网

地址：https://pytorch.org/get-started/locally/

打开你anaconda创建的虚拟环境

终端输入:

1

conda activate yolov5

进入环境后将上面框里的指令粘贴到终端里面

pytorch下载可能会超时导致下载失败，通过以下指令修改

下面的指令将连接超时修改成40s，读取超时的时间修改为100s

1
2

conda config --set remote_connect_timeout_secs 40
conda config --set remote_read_timeout_secs 100

安装完成后，在虚拟环境中输入

1

python

进入python，然后导入pytorch，验证是否安装正确

1
2

import torch
torch.cuda.is_available()

如果输出True，则成功安装

2.将环境导入pycharm

打开pycharm（ubuntu如何下载pycharm）

成功添加则右下角的解释器python版本后面会有环境名