0
点赞
收藏
分享

微信扫一扫

ejs的使用

你的益达233 2024-08-22 阅读 26

1、选择anaconda prompt

2、创建虚拟环境

3、激活进入虚拟环境

4、安装pytorch

怎么得到上面的这串命令?

输入nvidia-smi,查看cuda的版本号为11.7

我这里选择安装cuda的版本号为11.3,满足向下兼容即可。

在安装深度学习环境时,要满足cuda,pytorch,torchvision,Python它们之间的兼容,具体可见下表,我这里选择的组合用红框标出来了。

 去到pytorch官网,找到相应的命令

5、查看环境下的package

6、验证cuda是否可以使用

显示True就可以了

常见问题:

1、创建虚拟环境时出现failed,参考下面blog

conda创建环境过程中出现“Solving environment: failed”报错的解决办法

我是用了方法二解决了这个问题。

2、安装Pytorch时出现failed

很有可能是版本不兼容,多试试。这个问题比较难,可以参考这篇blog

Anaconda安装及配置(CPU/GPU)pytorch深度学习环境(超详细!!!)

补充:

1、查看环境列表

2、删除之前创建的虚拟环境

3、CUDA,GPU,Pytorch是什么及它们之间的关系

举报

相关推荐

0 条评论