装深度学习环境整个有很多教程,我主要说一说一个坑:
首先pytorch是分为gpu版本(cuda)和cpu版本的(cpu)
可能你已经在疑惑了,那怎么区分呢?在名字上都是 conda install pytorch呀
如果你打开pytorch的官网,官网上有分的很清楚
也就是说,在装pytorch时,会给你装一个cudatoolkit或者cpuonly模块来指定需要装什么版本的pytorch。
如果默认装pytorch时,即使指定了cudatoolkit,还是装的cpu版本的模块,那很有可能就是之前装了一个cpuonly模块
你需要conda list看一下,如果有,conda uninstall cpuonly就好啦!
如果一直卸不掉(conda在执行操作时会查看包的依赖关系,在卸载cpuonly模块时会额外的慢),可以使用pip卸载torch模块后,执行conda update --all,会自动将cpuonly模块卸了
安装pytorch默认安装cpu/torch.cuda.is_available()为False解决方法
最新推荐文章于 2024-07-23 08:25:13 发布