随笔分类 -  pytorch

摘要:conda create -n torch python=3.12 cuda-toolkit=12 cudnn cuda版本可以根据nvidia-smi查看,conda会自动配好各个模块的版本依赖。 cuda版的torch还是需要单独下载,conda直接从清华云下载安装的有问题。 可以从阿里云镜像下 阅读全文
posted @ 2025-04-25 14:46 科学修行的红客 阅读(70) 评论(0) 推荐(0)
摘要:pip install apex安装的不是nvidia的apex 需要到https://github.com/NVIDIA/apex下载,如果github打不开,可以用gettee拉取到国内下载 解压apex,进入apex文件夹,输入 pip install -v --disable-pip-ver 阅读全文
posted @ 2022-05-19 15:30 科学修行的红客 阅读(2868) 评论(0) 推荐(0)
摘要:最近做text to image相关研究,又来搞torch和cuda了,之前是直接用conda装的,用conda算是比较简单的了,这次尝试手动安装。 手动安装步骤: 1.先看项目的pytorch版本要求,然后到pytorch官网看torch和cuda和系统匹配的版本https://pytorch.o 阅读全文
posted @ 2022-05-08 21:11 科学修行的红客 阅读(280) 评论(0) 推荐(0)