随笔分类 -  cuda

摘要: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 科学修行的红客 阅读(2679) 评论(0) 推荐(0) 编辑
摘要:最近做text to image相关研究,又来搞torch和cuda了,之前是直接用conda装的,用conda算是比较简单的了,这次尝试手动安装。 手动安装步骤: 1.先看项目的pytorch版本要求,然后到pytorch官网看torch和cuda和系统匹配的版本https://pytorch.o 阅读全文
posted @ 2022-05-08 21:11 科学修行的红客 阅读(249) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示