小白深度学习环境搭建中的CUDA版本问题

        以pytorch=1.6.0,cuda10.2版本为例,其明确要求了cuda的版本为10.2,而我的服务器的cuda版本为11.7,明显的不符合,而重新安装cuda,服务器中多个cuda版本中共存,不仅费时费力,而且占用大量空间。服务器配置如下:

小白深度学习环境搭建中的CUDA版本问题_第1张图片

        根据网上查阅的资料显示,在大多数情况下,仅对cudatoolkit的版本进行安装 是可以满足 Pytorch 等框架的使用需求的。(官网上的安装命令中直接含有cudatoolkit版本的安装)

conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.2 -c pytorch

        另外,如果纠结于当前环境搭建所遇到的错误是否为cuda问题,可以通过查看当前安装torch的版本,对应cuda版本时候合适来判断。

小白深度学习环境搭建中的CUDA版本问题_第2张图片

 

        

 

你可能感兴趣的:(Deep,learning,深度学习,pytorch,python)