赞
踩
进入计算机管理,按图示步骤更新。
nvidia-smi
方法二:桌面→右键→NVIDIA控制面板→系统信息→组件→CUDA
我们需要的CUDA版本应 ≤ v11.6
一般不建议直接安装最高版本的CUDA,因为如果后续使用PyTorch或TensorFlow,95%的情况下是不兼容的(咱就是说稳定的总比最新的好)
我们需要的CUDA版本应 ≥ v11.0
这部分按部就班(记住安装路径,后面要用到)
nvcc -V
conda create -n mypytorch python=3.7
conda activate mypytorch
conda install cudatoolkit
pip install torch1.9.0+cu111 torchvision0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
(mypytorch) C:\Users\mysel>python
......................................
>>> import torch
>>> torch.__version__
>>> torch.cuda.is_available() ## 一定要输出True才是成功
————————————————
版权声明:本文为CSDN博主「zoxiii」的原创文章,遵循CC 4.0 BY-SA版权协议。
原文链接:第五步:RTX 3060配置CUDA和cuDNN、安装PyTorch
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。