前两天买的y7000p
到了,然后就要重新配下环境。
流程如下
- 首先下载
miniconda
,我下的是python3.8
的 - 创建自己的自定义环境
- 检查自己的
cuda
版本,我的是cuda:12.0
- 然后再
pytorch
上找到对应cuda
版本的进行下载,pip install
或者conda install
都可以
1.
直达链接
2.
conda create --name you_env_name python=3.8
conda activate you_env_name
3.
在cmd里输入 nvidia-smi 查看对应的 cuda 版本
4.
这里我们下载11.8 因为一般来说cuda会向下兼容的,但是不要差太多,拿捏不准可以都试试看。
直达链接
5.
测试下载好的 torch 是否可以使用cuda文章来源:https://www.toymoban.com/news/detail-790518.html
最后,测试完毕,没有问题,不过第一次 import torch可能时间要长一点,之后就很快了。文章来源地址https://www.toymoban.com/news/detail-790518.html
到了这里,关于cuda12.0 安装 pytorch的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!