建议自己阅读README文件。
1. 使用的是python 3.10版本。
conda create --name open-iad python=3.10
2.安装CUDA和pytorch-gpu版本
2.1安装CUDA
2.1.1先查看一下自己的所需CUDA版本(在NVIDIA控制面板中系统信息的组件查看自己所需的版本)。
还可以在终端用以下命令查看CUDA版本:
NVIDIA-SMI
2.1.2去官网上安装相应的CUDA版本(安装过程可以看别的博客有详细的安装过程)
CUDA:kCUDA Toolkit Archive | NVIDIA Developer
cuDNN: cuDNN 9.2.1 Downloads | NVIDIA Developer
安装即可。
2.2安装pytorch-gpu
对应着CUDA版本安装pytorch版本
可以pip下载(pip安装时不要用国内源否则下载的是cpu版本)不过一般下载时间慢(推荐下面的方式下载):
首先 先安装迅雷(不是打广告),迅雷下载国外的东西快,有节点。
第二步 在download.pytorch.org/whl/cu121/下载自己所需的torch,torchvision,torchaudio版本(文件形式whl)。
下载完以后 在自己的环境中安装whl文件
pip install some_package.whl
3.安装剩余的第三方库(安装时可以把后面的版本号删掉)
pip install -r requirements.txt
可以加国内源
国内源地址参考: pip国内镜像源-CSDN博客
如有侵权可以联系我。
要有问题可以私信我。
标签:open,iad,配置,CUDA,版本,NVIDIA,pip,安装,下载 From: https://blog.csdn.net/2301_80360664/article/details/140244367