踩了几乎能踩的所有坑,真是要命。
直接从1 购买云服务器开始看即可
一些可能用到的命令
-
nvidia-smi
:查看cuda driver的版本 -
看整个系统当前cuda的版本
nvcc --version
:说没有找到nvcc命令也不要着急下载nvcc,可以先用3.找找看cat /usr/local/cuda/version.txt
-
ls /usr/local/cuda*
:找找有没有下载过cuda,找到之后手动写到.bashrc文件里 -
python -c "import torch;print(torch.version.cuda)"
-
一些虚拟环境下的命令
import torch print(torch.__version__)# 当前环境下pytorch的版本 print(torch.cuda.is_available())# 当前环境下pytorch和cuda版本是否匹 配:true print(torch.version.cuda)# 当前环境下的cuda版本 print(torch.backends.cudnn.version())# 当前环境下的cudnn版本 import tensorflow as tf print(tf.test.is_built_with_cuda()) # 当前环境下tensorfolw和cuda版本是否匹配:true print(tf.test.is_gpu_available(cuda_only=False,min_cuda_compute_capab ility=None)) # gpu是否可以被tf调用:true
-
cat /usr/local/cuda