前两天买的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
最后,测试完毕,没有问题,不过第一次 import torch可能时间要长一点,之后就很快了。