CUDA安装步骤注意事项
cuda安装过程中出现两次安装路径(打开安装包出现一次,中间安装步骤中有一次安装位置),切记不要设置为同一路径,否则会清空安装文件,导致nvcc -V命令无效。
CUDA下载网址:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
cudnn下载网址:cuDNN Archive | NVIDIA Developer
anaconda prompt 进入环境:activate 环境名
设置清华园加速:pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
CUDA12.3安装pytorch:
conda install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=12.1 -c pytorch -c nvidia