conda环境下print(torch.cuda.is_available()) False

情景是:明明pip install torch>=1.10 安装成功了,但conda环境下运行


print(torch.cuda.is_available())


却得到


False


后来通过


print(torch.__version__)
确实是安装提CPU版本而非GPU版本



找到以前另一个安装笔记使用以下安装解决


conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia


但正确的处理方法应该是Pip手工安装 ,指定清楚,比如以下demo

pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio===0.7.2
速度慢还可以用迅雷下载实际文件后 直接安装。


参考自:https://blog.csdn.net/qq_46126258/article/details/112708781


猜你喜欢

解决向vim粘贴代码出现汉字乱码的问题

解决向vim粘贴代码出现汉字乱码的问题

方法分为永久解决方法和临时解决方法 1、临时解决方法 检查:在vim编辑器上 :set enc #得到 encoding=latin1 处理:在vim编辑器上 :set encoding=utf-8...

linux centos 解决卸载分区出现umount: target is busy.

linux centos 解决卸载分区出现umount: target is busy.

服务器一个分区的文件数过多,磁盘满,用rm时间太久,由于文件属于Cache可删除,决定直接格式化这个分区。 根据步骤,要选umount卸载分区,再mkfs.ext4格式化,再重新mount挂载。 【...

PHP CURL POST请求返回 400 Bad Request的处理

某套代码,都是linux,只是换了一台,换了php环境。 运行CURL POST请求返回 400 Bad Request。 经过验证排查,原因虽然使用POST,但我们的参数是在URL后边拼接的。 解决...