0
点赞
收藏
分享

微信扫一扫

Linux PyTorch安装配置

黄昏孤酒 2022-03-12 阅读 48

遇到的问题

在某一个虚拟环境test中通过pip uninstall torch后,发现所有的虚拟环境中的torch都被卸载,再次使用conda install安装(指定版本一样的问题),虽然可以安装,但是安装后在虚拟环境中无法找到torch及对应的包,也无法导入。

解决方法

查看显卡驱动版本号
在PyTorch官网中,通过同时指定torch版本以及服务器cuda对应的版本进行安装,解决!

指定版本
查看是否安装成功

举报

相关推荐

0 条评论