本以为是CUDA与torch的版本不对应,后来发现实际上安装的CUDA是11.0版本,但是通过命令行nvcc -V获得的cuda版本是10.1
通过以下的帖子获取解决办法
https://www.cnblogs.com/hf19950918/p/14707662.html
同时附上在同一台电脑中不同版本的CUDA,通过软连接切换的方法
本以为是CUDA与torch的版本不对应,后来发现实际上安装的CUDA是11.0版本,但是通过命令行nvcc -V获得的cuda版本是10.1
通过以下的帖子获取解决办法
https://www.cnblogs.com/hf19950918/p/14707662.html
同时附上在同一台电脑中不同版本的CUDA,通过软连接切换的方法