CUDA安装过程及pytorch操作记录

    科技2024-07-24  19

    如上图查看GPU支持的CUDA版本,去官网下载安装,过程如:

    https://blog.csdn.net/weixin_40994552/article/details/88087242?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-2.add_param_isCf&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-2.add_param_isCf

    CUDA安装成功检查:cmd命令提示符: nvcc -V

    从Anaconda Prompt 进入python环境 import torch,torch.cuda.is_available() 【此处available要加括号!!不然永远见不到True】查看cuda是否能用了。

    但此处一直False,CUDA由上一步显示成功,那么问题可能就在pytorch上了。

    后参考https://blog.csdn.net/dysljxjsq/article/details/104159802,其中说用conda命令下载的,是CPU版本,然后去查pytorch官网:https://pytorch.org/

    第一幅图代码中的(-c)应该能说明上位大神的情况吧(这里不能确定)

    因此改用pip安装就可以了

    https://blog.csdn.net/trinkled/article/details/105644017?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-1.add_param_isCf&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-1.add_param_isCf

    Processed: 0.022, SQL: 8