修复torch.cuda.is_available()返回Fal问题
学习官⽅60分钟教程时,尝试把张量移到GPU
tensor = torch.rand(3, 4)
# We move our tensor to the GPU if available
任弼时简介
if torch.cuda.is_available():
tensor = ('cuda')
print(f"Shape of tensor: {tensor.shape}")
print(f"Datatype of tensor: {tensor.dtype}")
print(f"Device tensor is stored on: {tensor.device}")
失败
Shape of tensor: torch.Size([3, 4])
Datatype of tensor: torch.float32
怀孕能用香水吗Device tensor is stored on: cpu
解决办法是按照规定的版本来配置环境。我的环境:
Win10
显卡:NVIDIA GeForce GTX 1060,驱动版本 457.30
pytorch:1.8.1+cu111
torchvision:0.9.1+cu111
torchaudio:0.8.1
CUDA:11.0
步骤:
1. 查看显卡⽀持哪些CUDA版本,我的显卡驱动最⾼可以升到466.47,最⾼⽀持CUDA 11.3。
2. 查看PyTorch官⽹,最⾼⽀持CUDA 11.1。对应显卡驱动要⼩于460.82,理论最⾼能⽤版本460.79。⽐我的版本新⼀个⽉,我就不更
新驱动了。
3. 安装pytorch。
pip安装pytorch
口语学习
小品大赛pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio===0.8.1 -f /whl/torch_stable.html
接近3GB
终于成功了
参考
⼀些错误原因
MacBook⽤不了CUDA
学前儿童艺术教育原因:MacBook不带显卡,CUDA 是 Nvidia专⽤的。
切换到台式机做训练。台式机也⽤不了GPU?
查看cuda版本,
nvcc -V滑冰动作
检查显卡驱动版本。如果找不到命令,那就要添加环境路径。
nvidia-smi
原因:两个CUDA版本不同。
解:安装11.1版本CUDA。如果torch.cuda.is_available()返回Ture,但查看cuda版本还是10.1,那么要修改nvcc的环境变量。
升级⼀下显卡驱动,对应的CUDA版本变成11.3了
原因:显卡驱动版本太⾼。
解:
重新安装历史版本的显卡驱动
换回457.30,还是不⾏。
⼀查,更新完pytorch,版本变得更旧了?
>>> print(torch.__version__)
智氏1.6.0
原因:⽤了这个命令安装pytorch
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge
解:换pip安装
pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio===0.8.1 -f /whl/torch_stable.html 参考
原因:pytorch最⾼⽀持CUDA11.1
原因:pytorch版本不对
查看pytorch版本
>>> print(torch.__version__)设计吧廊
1.8.1+cpu