Linux PyTorch安装配置

阅读 57

2022-03-12

遇到的问题

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

解决方法

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

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

精彩评论(0)

0 0 举报