编辑
评论
收藏
点赞
分享
当你的系统拥有多个版本cuda时,想切换你想用的版本的时,有两种方法
方法1:临时修改,在cmd命令行进行以下操作
set CUDA_HOME="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8" set PATH=%CUDA_HOME%\bin;%CUDA_HOME%\libnvvp;%PATH% nvcc --version
方法2:环境变量只保留你想要的那个版本
如何查看当前的cuda版本
nvcc --version
比较文档的cuda与pytorch
pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
把好文章收藏到微信
打开微信,扫码查看
请使用手机微信扫一扫扫描二维码支付
请使用手机支付宝扫一扫扫描二维码支付
还没有账号?立即注册
评论