随笔分类 - pytorch
摘要:conda create -n torch python=3.12 cuda-toolkit=12 cudnn cuda版本可以根据nvidia-smi查看,conda会自动配好各个模块的版本依赖。 cuda版的torch还是需要单独下载,conda直接从清华云下载安装的有问题。 可以从阿里云镜像下
阅读全文
摘要:pip install apex安装的不是nvidia的apex 需要到https://github.com/NVIDIA/apex下载,如果github打不开,可以用gettee拉取到国内下载 解压apex,进入apex文件夹,输入 pip install -v --disable-pip-ver
阅读全文
摘要:最近做text to image相关研究,又来搞torch和cuda了,之前是直接用conda装的,用conda算是比较简单的了,这次尝试手动安装。 手动安装步骤: 1.先看项目的pytorch版本要求,然后到pytorch官网看torch和cuda和系统匹配的版本https://pytorch.o
阅读全文